Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,12 @@
|
|
| 1 |
import os
|
| 2 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
os.system("nvcc --version")
|
| 4 |
print(os.environ.get('CUDA_PATH'))
|
| 5 |
os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
|
|
|
|
| 1 |
import os
|
| 2 |
import os
|
| 3 |
+
|
| 4 |
+
os.system("wget https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64 -O cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb")
|
| 5 |
+
os.system("dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb")
|
| 6 |
+
os.system("apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub")
|
| 7 |
+
os.system("apt-get update")
|
| 8 |
+
os.system("apt-get install cuda")
|
| 9 |
+
|
| 10 |
os.system("nvcc --version")
|
| 11 |
print(os.environ.get('CUDA_PATH'))
|
| 12 |
os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
|