update
Browse files
app.py
CHANGED
|
@@ -58,6 +58,7 @@ importlib.invalidate_caches()
|
|
| 58 |
diff_gaussian_rasterization = importlib.import_module("diff_gaussian_rasterization")
|
| 59 |
|
| 60 |
os.system('pip install --no-deps dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html')
|
|
|
|
| 61 |
# os.system('conda install conda-forge::ffmpeg')
|
| 62 |
|
| 63 |
print(torch.__version__)
|
|
|
|
| 58 |
diff_gaussian_rasterization = importlib.import_module("diff_gaussian_rasterization")
|
| 59 |
|
| 60 |
os.system('pip install --no-deps dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html')
|
| 61 |
+
os.system('pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu124')
|
| 62 |
# os.system('conda install conda-forge::ffmpeg')
|
| 63 |
|
| 64 |
print(torch.__version__)
|