test

app.py

@@ -93,10 +93,16 @@ def upsample_features(image, model_option):
     with torch.no_grad():
         from setuptools import setup, find_packages
         from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtension
-        from torch.utils.cpp_extension import _find_cuda_home
 
-        os.environ['CUDA_HOME'] = _find_cuda_home()
-        print(os.environ['CUDA_HOME'])
+        cuda_available = torch.cuda.is_available()
+        cuda_version = torch.version.cuda
+
+        if cuda_available:
+            print(f"CUDA is available. CUDA Version: {cuda_version}")
+            print(f"CUDA Device Name: {torch.cuda.get_device_name(0)}")
+        else:
+            print("CUDA is not available.")
+
         setup(
             name='featup',
             version='0.1.2',