Try new lib setup

demo/gs_train.py

@@ -132,7 +132,19 @@ def train(
         densify_grad_threshold=densify_grad_threshold,
         random_background=random_background
     )
-
+    try:
+        import subprocess
+        nvcc_version = subprocess.check_output(['nvcc', '--version']).decode('utf-8')
+        print("NVCC Driver Version:", nvcc_version)
+    except Exception as e:
+        print("Error fetching NVCC Driver Version:", e)
+
+    try:
+        import sys
+        python_version = sys.version
+        print("Python Version:", python_version)
+    except Exception as e:
+        print("Error fetching Python Version:", e)
 
     #
     # Copyright (C) 2023, Inria

requirements.txt

@@ -1,5 +1,5 @@
 # git+https://github.com/ostapagon/wild-gaussian-splatting.git@f9a871b1aa25709befe8ddb57aaacdf3f7a31c2f#egg=wild-gaussian-splatting
---extra-index-url https://download.pytorch.org/whl/cu121
+--extra-index-url https://download.pytorch.org/whl/cu124
 
 torch==2.4.0
 torchvision==0.19.0
@@ -14,10 +14,12 @@ trimesh
 
 https://huggingface.co/spaces/ostapagon/mast3r-3dgs/resolve/main/wheels/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
 https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
+https://huggingface.co/spaces/ostapagon/mast3r-3dgs/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
+
+
 tensorboard
 pyglet<2
 roma
-https://huggingface.co/spaces/ostapagon/mast3r-3dgs/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
 camtools==0.1.4
 einops==0.7.0
 lpips==0.1.4

wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e7c7ac2bf4138985e267a1973ae3eaae6a8f5e5226edfd98ba40d3125485570a
-size 3325550
+oid sha256:70a25a2c4e6b23bab1b403e04f12dbbbe0ac0aa0cd67c04d1345c1a51f3a4d4e
+size 3290301

wheels/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d4c8abe48b59dad3570ea1e4d5540303f8a16869d5a61a2737b043142c220497
-size 3082355
+oid sha256:6da49f22b9fd37ee6ed579a9724670fc54c0514fd418931b486996c0c2ff8748
+size 3041144