new
Browse files- app.py +2 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -32,8 +32,9 @@ import tempfile
|
|
| 32 |
from huggingface_hub import hf_hub_download
|
| 33 |
|
| 34 |
def install_cuda_toolkit():
|
| 35 |
-
CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
|
| 36 |
# CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
|
|
|
|
| 37 |
print(f"==> install cuda {CUDA_TOOLKIT_URL}")
|
| 38 |
CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
|
| 39 |
subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
|
|
|
|
| 32 |
from huggingface_hub import hf_hub_download
|
| 33 |
|
| 34 |
def install_cuda_toolkit():
|
| 35 |
+
# CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
|
| 36 |
# CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
|
| 37 |
+
CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
|
| 38 |
print(f"==> install cuda {CUDA_TOOLKIT_URL}")
|
| 39 |
CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
|
| 40 |
subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
|
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
torch==2.1.0
|
| 2 |
torchvision==0.16.0
|
| 3 |
torchaudio==2.1.0
|
| 4 |
-
-f https://download.pytorch.org/whl/
|
| 5 |
|
| 6 |
pytorch-lightning==2.1.2
|
| 7 |
gradio==3.41.2
|
|
|
|
| 1 |
torch==2.1.0
|
| 2 |
torchvision==0.16.0
|
| 3 |
torchaudio==2.1.0
|
| 4 |
+
-f https://download.pytorch.org/whl/cu121
|
| 5 |
|
| 6 |
pytorch-lightning==2.1.2
|
| 7 |
gradio==3.41.2
|