install torch

app.py

@@ -17,6 +17,7 @@ def install_cuda_toolkit():
     )
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
 install_cuda_toolkit()
+os.system('pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124')
 os.system("pip list | grep torch")
 os.system('nvcc -V')
 print("cd /home/user/app/step1x3d_texture/differentiable_renderer/ && python setup.py install")

requirements.txt

@@ -1,6 +1,3 @@
-torch==2.5.1+cu124
-torchvision==0.20.1+cu124
-torchaudio==2.5.1+cu124
 datasets==2.19.0
 diffusers==0.32.2
 einops==0.8.0