zhiweili
commited on
Commit
·
9114cb3
1
Parent(s):
7966432
show cuda
Browse files- app_onediff.py +1 -0
app_onediff.py
CHANGED
|
@@ -45,6 +45,7 @@ def runcmd(cmd, verbose = False, *args, **kwargs):
|
|
| 45 |
|
| 46 |
print(os.environ.get('CUDA_PATH'))
|
| 47 |
print(os.environ.get('CUDA_HOME'))
|
|
|
|
| 48 |
|
| 49 |
os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")
|
| 50 |
|
|
|
|
| 45 |
|
| 46 |
print(os.environ.get('CUDA_PATH'))
|
| 47 |
print(os.environ.get('CUDA_HOME'))
|
| 48 |
+
os.system("pip show cuda")
|
| 49 |
|
| 50 |
os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")
|
| 51 |
|