update

app.py

@@ -57,8 +57,8 @@ diff_gaussian_rasterization = importlib.import_module("diff_gaussian_rasterizati
 
 subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-deps", "dgl", "-f", "https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html"])
 
-subprocess.check_call([sys.executable, "-m", "pip", "uninstall", "-y", "torch", "torchvision", "torchaudio"])
-subprocess.check_call([sys.executable, "-m", "pip", "install", "torch==2.4.0", "torchvision==0.19.0", "torchaudio==2.4.0", "--index-url", "https://download.pytorch.org/whl/cu124"])
+# subprocess.check_call([sys.executable, "-m", "pip", "uninstall", "-y", "torch", "torchvision", "torchaudio"])
+# subprocess.check_call([sys.executable, "-m", "pip", "install", "torch==2.4.0", "torchvision==0.19.0", "torchaudio==2.4.0", "--index-url", "https://download.pytorch.org/whl/cu124"])
 # os.system('conda install conda-forge::ffmpeg')
 
 site.main()  # re-processes every *.pth in site-packages

requirements.txt

@@ -23,6 +23,6 @@ timm
 omegaconf
 hydra-core
 wandb
-# torch==2.4.0 --index-url https://download.pytorch.org/whl/cu124
+torch==2.4.0 --index-url https://download.pytorch.org/whl/cu124
 # torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu124
 # torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu124