alexnasa commited on
Commit
b354162
·
verified ·
1 Parent(s): 3d9fee9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -31,9 +31,9 @@ import subprocess
31
  def install_cuda_toolkit():
32
  CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.6.0/local_installers/cuda_12.6.0_560.28.03_linux.run"
33
  CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
34
- subprocess.check_call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE], shell=True)
35
- subprocess.check_call(["chmod", "+x", CUDA_TOOLKIT_FILE], shell=True)
36
- subprocess.check_call([CUDA_TOOLKIT_FILE, "--toolkit"], shell=True)
37
 
38
  os.environ["CUDA_HOME"] = "/usr/local/cuda"
39
  os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
 
31
  def install_cuda_toolkit():
32
  CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.6.0/local_installers/cuda_12.6.0_560.28.03_linux.run"
33
  CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
34
+ subprocess.check_call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
35
+ subprocess.check_call(["chmod", "+x", CUDA_TOOLKIT_FILE])
36
+ subprocess.check_call([CUDA_TOOLKIT_FILE, "--toolkit"])
37
 
38
  os.environ["CUDA_HOME"] = "/usr/local/cuda"
39
  os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])