zhiweili
commited on
Commit
·
e9f9702
1
Parent(s):
5ea0c6e
test nvcc
Browse files- app_onediff.py +2 -0
app_onediff.py
CHANGED
|
@@ -46,6 +46,8 @@ def runcmd(cmd, verbose = False, *args, **kwargs):
|
|
| 46 |
print(os.environ.get('CUDA_PATH'))
|
| 47 |
print(os.environ.get('CUDA_HOME'))
|
| 48 |
os.system("pip show torch")
|
|
|
|
|
|
|
| 49 |
|
| 50 |
# os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")
|
| 51 |
|
|
|
|
| 46 |
print(os.environ.get('CUDA_PATH'))
|
| 47 |
print(os.environ.get('CUDA_HOME'))
|
| 48 |
os.system("pip show torch")
|
| 49 |
+
os.system("nvcc --version")
|
| 50 |
+
os.system("which nvcc")
|
| 51 |
|
| 52 |
# os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")
|
| 53 |
|