pytorch latest in conda

Signed-off-by: Mikael Bourges-Sevenier <mikael.sevenier@sima.com>
quic · May 1, 2021 · 8deef76 · 8deef76
1 parent 527504f
commit 8deef76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packaging/environment.yml b/packaging/environment.yml
@@ -20,6 +20,6 @@ dependencies:
   - tensorflow=1.15
   - libprotobuf=3.8
   - mkl-devel
-  - pytorch=1.8
+  - pytorch
   - torchvision
   - cudatoolkit>=10