Spaces:
Running
on
Zero
Running
on
Zero
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([
|