ReubenSun commited on
Commit
7c6dd01
·
1 Parent(s): e227e15
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -17,7 +17,6 @@ def install_cuda_toolkit():
17
  )
18
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
19
  install_cuda_toolkit()
20
- os.system('pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124')
21
  os.system("pip list | grep torch")
22
  os.system('nvcc -V')
23
  print("cd /home/user/app/step1x3d_texture/differentiable_renderer/ && python setup.py install")
 
17
  )
18
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
19
  install_cuda_toolkit()
 
20
  os.system("pip list | grep torch")
21
  os.system('nvcc -V')
22
  print("cd /home/user/app/step1x3d_texture/differentiable_renderer/ && python setup.py install")