yslan commited on
Commit
ce56ec2
·
1 Parent(s): 0ef9765
Files changed (1) hide show
  1. app.py +1 -1
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([