Spaces:

KumaPower
/

AvatarArtist

Running on Zero

刘虹雨 commited on Mar 31

Commit

78b73ce

1 Parent(s): ab4d9b9

update code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ import sys
 os.environ["MEDIAPIPE_DISABLE_GPU"] = "1"  # Disable GPU for MediaPipe
 def install_cuda_toolkit():
-    CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
     print(f"[INFO] Downloading CUDA Toolkit from {CUDA_TOOLKIT_URL} ...")

 os.environ["MEDIAPIPE_DISABLE_GPU"] = "1"  # Disable GPU for MediaPipe
 def install_cuda_toolkit():
+    CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
     print(f"[INFO] Downloading CUDA Toolkit from {CUDA_TOOLKIT_URL} ...")