Spaces:
Running
on
Zero
Running
on
Zero
刘虹雨
commited on
Commit
·
101ef26
1
Parent(s):
4c94a4e
update code
Browse files
app.py
CHANGED
|
@@ -967,7 +967,9 @@ if __name__ == '__main__':
|
|
| 967 |
else:
|
| 968 |
logging.info("Detected CUDA: %s" % _get_output(["nvcc", "--version"]))
|
| 969 |
|
| 970 |
-
|
|
|
|
|
|
|
| 971 |
launch_pretrained()
|
| 972 |
|
| 973 |
|
|
|
|
| 967 |
else:
|
| 968 |
logging.info("Detected CUDA: %s" % _get_output(["nvcc", "--version"]))
|
| 969 |
|
| 970 |
+
print("CUDA_HOME =", os.environ.get("CUDA_HOME"))
|
| 971 |
+
from torch.utils.cpp_extension import CUDA_HOME
|
| 972 |
+
print("CUDA_HOME from PyTorch:", CUDA_HOME)
|
| 973 |
launch_pretrained()
|
| 974 |
|
| 975 |
|