Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ def install_cuda_toolkit():
|
|
41 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
42 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
43 |
print("==> finfish install")
|
44 |
-
|
45 |
@spaces.GPU
|
46 |
def check_gpu():
|
47 |
os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
|
|
|
41 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
42 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
43 |
print("==> finfish install")
|
44 |
+
install_cuda_toolkit()
|
45 |
@spaces.GPU
|
46 |
def check_gpu():
|
47 |
os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
|