env version for lightning

divyansh7877 · Aug 8, 2022 · 0c29bdf · 0c29bdf
1 parent 08baf16
commit 0c29bdf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/environment.yml b/environment.yml
@@ -16,10 +16,10 @@ dependencies:
   - pillow
   - scikit-image
   - pip:
-    - pytorch-lightning
+    - pytorch-lightning==1.6.5
     - opencv-contrib-python
     - albumentations
     - timm==0.6.5
     - test-tube
-    - --find-links https://github.com/pytorch/TensorRT/releases
-    - torch-tensorrt
+    # - --find-links https://github.com/pytorch/TensorRT/releases
+    # - torch-tensorrt