abreza commited on
Commit
63b27a9
·
1 Parent(s): ae51641
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -106,6 +106,7 @@ def upsample_features(image, model_option):
106
  from setuptools import setup, find_packages
107
  from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtension
108
 
 
109
  cuda_home = find_cuda_home()
110
  if cuda_home:
111
  print(cuda_home)
 
106
  from setuptools import setup, find_packages
107
  from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtension
108
 
109
+ print(subprocess.check_output(['ls', '/usr/local/']).decode())
110
  cuda_home = find_cuda_home()
111
  if cuda_home:
112
  print(cuda_home)