Spaces:
Runtime error
Runtime error
update
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ def install_cuda_toolkit():
|
|
| 25 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
| 26 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
| 27 |
|
| 28 |
-
install_cuda_toolkit() # to compile the dependencies
|
| 29 |
|
| 30 |
pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
|
| 31 |
version_str="".join([
|
|
|
|
| 25 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
| 26 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
| 27 |
|
| 28 |
+
# install_cuda_toolkit() # to compile the dependencies
|
| 29 |
|
| 30 |
pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
|
| 31 |
version_str="".join([
|