test
Browse files
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)
|