Spaces:
Running
on
A100
Running
on
A100
check cuda
Browse files
app.py
CHANGED
@@ -192,6 +192,13 @@ def make_mesh(mesh_fpath, planes):
|
|
192 |
@spaces.GPU
|
193 |
def make3d(input_image, sample_steps, sample_seed):
|
194 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
195 |
global model
|
196 |
if IS_FLEXICUBES:
|
197 |
model.init_flexicubes_geometry(device)
|
|
|
192 |
@spaces.GPU
|
193 |
def make3d(input_image, sample_steps, sample_seed):
|
194 |
|
195 |
+
cuda_path = find_cuda()
|
196 |
+
|
197 |
+
if cuda_path:
|
198 |
+
print(f"CUDA installation found at: {cuda_path}")
|
199 |
+
else:
|
200 |
+
print("CUDA installation not found")
|
201 |
+
|
202 |
global model
|
203 |
if IS_FLEXICUBES:
|
204 |
model.init_flexicubes_geometry(device)
|