Add nvidia-cuda-runtime + LD_LIBRARY_PATH shim so nvdiffrast's JIT-compiled CUDA extension can find libcudart.so.13 on Blackwell/CUDA-13 ZeroGPU workers

app.py

@@ -13,6 +13,16 @@ from glob import glob
 import requests
 from huggingface_hub import snapshot_download
 
+# nvdiffrast JIT-compiles a native CUDA extension against the container's system
+# CUDA toolkit; on ZeroGPU workers with a driver reporting CUDA 13 (e.g. Blackwell
+# MIG slices), libcudart.so.13 isn't on the default linker search path, so pull it
+# in from the nvidia-cuda-runtime pip package instead.
+import sysconfig
+
+_cu13_lib_dir = os.path.join(sysconfig.get_paths()["purelib"], "nvidia", "cu13", "lib")
+if os.path.isdir(_cu13_lib_dir):
+    os.environ["LD_LIBRARY_PATH"] = _cu13_lib_dir + os.pathsep + os.environ.get("LD_LIBRARY_PATH", "")
+
 # Download models
 os.makedirs("ckpts", exist_ok=True)
 

requirements.txt

@@ -1,5 +1,6 @@
 spaces
 setuptools<82
+nvidia-cuda-runtime
 torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121