new

app.py

@@ -1,5 +1,8 @@
 import spaces
 import os
+os.environ["CXX"] = os.popen("which g++").read().strip()
+os.environ["CC"] = os.popen("which gcc").read().strip()
+
 import imageio
 import numpy as np
 import torch
@@ -29,9 +32,9 @@ import tempfile
 from huggingface_hub import hf_hub_download
 
 def install_cuda_toolkit():
-    print("==> install cuda 118")
     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_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
+    print(f"==> install cuda {CUDA_TOOLKIT_URL}")
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
     subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
     subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])