kaifz commited on
Commit
d7da98a
·
1 Parent(s): 4f34a99
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -57,9 +57,12 @@ site.main() # re-processes every *.pth in site-packages
57
  importlib.invalidate_caches()
58
  diff_gaussian_rasterization = importlib.import_module("diff_gaussian_rasterization")
59
 
60
- os.system('pip install dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html')
61
  # os.system('conda install conda-forge::ffmpeg')
62
 
 
 
 
63
  import sys
64
  sys.path.insert(0, str(Path(__file__).parent / "src"))
65
  sys.path.append(str(Path(__file__).parent / "src" / "experiments"))
 
57
  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__)
64
+ print(torch.version.cuda)
65
+
66
  import sys
67
  sys.path.insert(0, str(Path(__file__).parent / "src"))
68
  sys.path.append(str(Path(__file__).parent / "src" / "experiments"))