gensym commited on
Commit
2385d44
·
1 Parent(s): ae973c3
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -16,6 +16,10 @@ from trellis.representations import Gaussian, MeshExtractResult
16
  from trellis.utils import render_utils, postprocessing_utils
17
 
18
 
 
 
 
 
19
  MAX_SEED = np.iinfo(np.int32).max
20
  TMP_DIR = "/tmp/Trellis-demo"
21
 
 
16
  from trellis.utils import render_utils, postprocessing_utils
17
 
18
 
19
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
20
+ # True
21
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
22
+
23
  MAX_SEED = np.iinfo(np.int32).max
24
  TMP_DIR = "/tmp/Trellis-demo"
25