JiantaoLin
commited on
Commit
·
50e89c5
1
Parent(s):
911fe42
new
Browse files
app.py
CHANGED
|
@@ -50,6 +50,8 @@ install_cuda_toolkit()
|
|
| 50 |
|
| 51 |
@spaces.GPU
|
| 52 |
def check_gpu():
|
|
|
|
|
|
|
| 53 |
os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
|
| 54 |
os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
|
| 55 |
# os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-12.1/lib64'
|
|
|
|
| 50 |
|
| 51 |
@spaces.GPU
|
| 52 |
def check_gpu():
|
| 53 |
+
if "CUDA_VISIBLE_DEVICES" in os.environ:
|
| 54 |
+
del os.environ["CUDA_VISIBLE_DEVICES"]
|
| 55 |
os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
|
| 56 |
os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
|
| 57 |
# os.environ['LD_LIBRARY_PATH'] += ':/usr/local/cuda-12.1/lib64'
|