Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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]
|
| 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"])
|
|
|
|
| 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"])
|