new
Browse files
app.py
CHANGED
|
@@ -30,8 +30,8 @@ from huggingface_hub import hf_hub_download
|
|
| 30 |
|
| 31 |
def install_cuda_toolkit():
|
| 32 |
print("==> install cuda 118")
|
| 33 |
-
CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
|
| 34 |
-
|
| 35 |
CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
|
| 36 |
subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
|
| 37 |
subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
|
|
|
|
| 30 |
|
| 31 |
def install_cuda_toolkit():
|
| 32 |
print("==> install cuda 118")
|
| 33 |
+
# CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
|
| 34 |
+
CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
|
| 35 |
CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
|
| 36 |
subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
|
| 37 |
subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
|