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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def install_cuda_toolkit():
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"])
 
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, "--silent", "--toolkit"])
37
 
38
  os.environ["CUDA_HOME"] = "/usr/local/cuda"
39
  os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])