Spaces:
Runtime error
Runtime error
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)
|