alexnasa commited on
Commit
4be5030
·
verified ·
1 Parent(s): 5bd591c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import os
5
  import subprocess
6
 
7
  def install_cuda_toolkit():
8
- CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
9
  CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
10
  subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
11
  subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
 
5
  import subprocess
6
 
7
  def install_cuda_toolkit():
8
+ CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.6.3/local_installers/cuda_12.6.3_560.35.05_linux.run"
9
  CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
10
  subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
11
  subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])