Spaces:
Build error
Build error
Pie31415
commited on
Commit
·
a28bde4
1
Parent(s):
d57512d
changed cuda version to 10.1
Browse files- app.py +0 -5
- requirements.txt +2 -2
app.py
CHANGED
|
@@ -7,11 +7,6 @@ import torch
|
|
| 7 |
import matplotlib.pyplot as plt
|
| 8 |
from PIL import Image
|
| 9 |
|
| 10 |
-
import subprocess, sys
|
| 11 |
-
subprocess.check_call(["pip", "install", "torch==1.6.0+cu101",
|
| 12 |
-
"torchvision==0.7.0+cu101",
|
| 13 |
-
"-f", "https://download.pytorch.org/whl/torch_stable.html"])
|
| 14 |
-
|
| 15 |
print(torch.__version__)
|
| 16 |
print(torch.version.cuda)
|
| 17 |
|
|
|
|
| 7 |
import matplotlib.pyplot as plt
|
| 8 |
from PIL import Image
|
| 9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
print(torch.__version__)
|
| 11 |
print(torch.version.cuda)
|
| 12 |
|
requirements.txt
CHANGED
|
@@ -6,6 +6,6 @@ kornia==0.4.0
|
|
| 6 |
numpy
|
| 7 |
matplotlib
|
| 8 |
pillow
|
| 9 |
-
torch==1.6.0
|
| 10 |
-
torchvision==0.7.0
|
| 11 |
easydict
|
|
|
|
| 6 |
numpy
|
| 7 |
matplotlib
|
| 8 |
pillow
|
| 9 |
+
torch==1.6.0+cu101
|
| 10 |
+
torchvision==0.7.0+cu101
|
| 11 |
easydict
|