Use the cpu version of torch and torchvision for tests

healthonrails · Dec 17, 2020 · e840147 · e840147
1 parent c024247
commit e840147
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,8 +1,9 @@
 pyyaml>=5.3
 easydict>=1.9
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==1.4.0+gpu
-torchvision==0.5.0
+torch==1.4.0+cpu
+-find-links https://download.pytorch.org/whl/torch_stable.html
+torchvision==0.5.0+cpu
 decord>=0.4.0
 opencv-python>=4.1.2.30
 opencv-contrib-python>=4.1.2.30