LTT commited on
Commit
0ab94ef
·
verified ·
1 Parent(s): 003affa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- # install_cuda_toolkit()
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'