changed cuda version to 10.1

app.py

@@ -7,11 +7,6 @@ import torch
 import matplotlib.pyplot as plt
 from PIL import Image
 
-import subprocess, sys
-subprocess.check_call(["pip", "install", "torch==1.6.0+cu101",
-                       "torchvision==0.7.0+cu101",
-                       "-f", "https://download.pytorch.org/whl/torch_stable.html"])
-
 print(torch.__version__)
 print(torch.version.cuda)
 

requirements.txt

@@ -6,6 +6,6 @@ kornia==0.4.0
 numpy
 matplotlib
 pillow
-torch==1.6.0
-torchvision==0.7.0
+torch==1.6.0+cu101
+torchvision==0.7.0+cu101
 easydict