JiantaoLin
commited on
Commit
Β·
730b619
1
Parent(s):
e784881
new
Browse files
app.py
CHANGED
|
@@ -87,7 +87,6 @@ k3d_wrapper = init_wrapper_from_config('./pipeline/pipeline_config/default.yaml'
|
|
| 87 |
from models.ISOMER.scripts.utils import fix_vert_color_glb
|
| 88 |
torch.backends.cuda.matmul.allow_tf32 = True
|
| 89 |
|
| 90 |
-
@spaces.GPU
|
| 91 |
def check_gpu():
|
| 92 |
os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
|
| 93 |
os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
|
|
@@ -101,7 +100,6 @@ def check_gpu():
|
|
| 101 |
print(f"Successfully preloaded {cuda_lib_path}")
|
| 102 |
except OSError as e:
|
| 103 |
print(f"Failed to preload {cuda_lib_path}: {e}")
|
| 104 |
-
print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
|
| 105 |
check_gpu()
|
| 106 |
|
| 107 |
TEMP_MESH_ADDRESS=''
|
|
|
|
| 87 |
from models.ISOMER.scripts.utils import fix_vert_color_glb
|
| 88 |
torch.backends.cuda.matmul.allow_tf32 = True
|
| 89 |
|
|
|
|
| 90 |
def check_gpu():
|
| 91 |
os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
|
| 92 |
os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
|
|
|
|
| 100 |
print(f"Successfully preloaded {cuda_lib_path}")
|
| 101 |
except OSError as e:
|
| 102 |
print(f"Failed to preload {cuda_lib_path}: {e}")
|
|
|
|
| 103 |
check_gpu()
|
| 104 |
|
| 105 |
TEMP_MESH_ADDRESS=''
|