lemonaddie commited on
Commit
a388382
1 Parent(s): b5b8960

Update app_recon.py

Browse files
Files changed (1) hide show
  1. app_recon.py +2 -0
app_recon.py CHANGED
@@ -9,6 +9,8 @@ import numpy as np
9
  import torch as torch
10
  from PIL import Image
11
 
 
 
12
  from gradio_imageslider import ImageSlider
13
  from bilateral_normal_integration.bilateral_normal_integration_cupy import bilateral_normal_integration_function
14
 
 
9
  import torch as torch
10
  from PIL import Image
11
 
12
+ print(torch.version.cuda)
13
+
14
  from gradio_imageslider import ImageSlider
15
  from bilateral_normal_integration.bilateral_normal_integration_cupy import bilateral_normal_integration_function
16