cuda
Browse files
app.py
CHANGED
|
@@ -14,6 +14,7 @@ def install_cuda_toolkit():
|
|
| 14 |
os.environ["CUDA_HOME"],
|
| 15 |
"" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
|
| 16 |
)
|
|
|
|
| 17 |
install_cuda_toolkit()
|
| 18 |
os.system('bash install.sh')
|
| 19 |
import time
|
|
|
|
| 14 |
os.environ["CUDA_HOME"],
|
| 15 |
"" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
|
| 16 |
)
|
| 17 |
+
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
| 18 |
install_cuda_toolkit()
|
| 19 |
os.system('bash install.sh')
|
| 20 |
import time
|