Update environment.yml

raghunandan-r · Feb 6, 2022 · 383d13c · 383d13c
1 parent 0c7b2b3
commit 383d13c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/code/environment.yml b/code/environment.yml
@@ -12,6 +12,8 @@ dependencies:
   - opencv
   - jupyterlab
   - ipywidgets
+  - pytorch=1.9.0
+  - torchvision=0.10.0
   - fastai=2.5.0
   - albumentations
   - tqdm
@@ -21,4 +23,4 @@ dependencies:
     - pyclothoids
     - pygame
     - imageio
-    - imageio-ffmpeg
+    - imageio-ffmpeg