Update environment-cuda.yml

facebookresearch · Dec 6, 2019 · 3aef510 · 3aef510
1 parent 01c9570
commit 3aef510
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/environment-cuda.yml b/environment-cuda.yml
@@ -6,7 +6,7 @@ channels:
 
 dependencies:
   - python=3.7
-  - cudatoolkit=10.0
+  - cudatoolkit=10
   - ffmpeg==4.2
   - pytorch=1.3.1
   - scipy==1.3.1