alexnasa commited on
Commit
5e494f2
·
verified ·
1 Parent(s): c53014b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -25
app.py CHANGED
@@ -4,31 +4,6 @@ import gc
4
  import os
5
  import subprocess
6
 
7
- def install_cuda_toolkit():
8
- CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.6.3/local_installers/cuda_12.6.3_560.35.05_linux.run"
9
- CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
10
- subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
11
- subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
12
- subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
13
-
14
- os.environ["CUDA_HOME"] = "/usr/local/cuda"
15
- os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
16
- os.environ["CPATH"] = "%s/include" % (os.environ["CUDA_HOME"])
17
- os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
18
- os.environ["CUDA_HOME"],
19
- "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
20
- )
21
- # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
22
- os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
23
- print("==> finished installation")
24
-
25
- install_cuda_toolkit()
26
-
27
- def sh(cmd): subprocess.check_call(cmd, shell=True)
28
-
29
- sh("pip install torch-scatter")
30
-
31
-
32
  import shutil
33
  import sys
34
  import tempfile
 
4
  import os
5
  import subprocess
6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  import shutil
8
  import sys
9
  import tempfile