lemonaddie commited on
Commit
fbe3aa1
1 Parent(s): 89febce

Update app_recon.py

Browse files
Files changed (1) hide show
  1. app_recon.py +2 -1
app_recon.py CHANGED
@@ -10,7 +10,8 @@ import torch as torch
10
  from PIL import Image
11
 
12
  print(torch.version.cuda)
13
- os.system('libcusolver.so.11')
 
14
 
15
  from gradio_imageslider import ImageSlider
16
  from bilateral_normal_integration.bilateral_normal_integration_cupy import bilateral_normal_integration_function
 
10
  from PIL import Image
11
 
12
  print(torch.version.cuda)
13
+ os.system('locate libcusolver.so.11')
14
+ print(os.environ[“CUDA_VISIBLE_DEVICES”])
15
 
16
  from gradio_imageslider import ImageSlider
17
  from bilateral_normal_integration.bilateral_normal_integration_cupy import bilateral_normal_integration_function