tokenid commited on
Commit
ac99742
1 Parent(s): 0cd8dc5

check cuda

Browse files
Files changed (1) hide show
  1. app.py +7 -0
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)