Spaces:

haodongli
/

LucidDreamer

Runtime error

haodongli commited on Oct 6, 2024

Commit

4b2844a

verified ·

1 Parent(s): 7a4d9f7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,9 @@ import gradio as gr
 import numpy as np
 import os
 try:
     from train import *
     print('==> simple-knn & diff-gaussian-rasterization already installed!')

 import numpy as np
 import os
+import torch
+print(torch.cuda.get_device_capability(0))
 try:
     from train import *
     print('==> simple-knn & diff-gaussian-rasterization already installed!')