Spaces:
Runtime error
Runtime error
update app
Browse files
app.py
CHANGED
|
@@ -18,6 +18,13 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 18 |
|
| 19 |
print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
|
| 20 |
print("torch.__version__ =", torch.__version__)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
print("Using device:", device)
|
| 22 |
|
| 23 |
from diffusers import FlowMatchEulerDiscreteScheduler
|
|
|
|
| 18 |
|
| 19 |
print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
|
| 20 |
print("torch.__version__ =", torch.__version__)
|
| 21 |
+
print("torch.version.cuda =", torch.version.cuda)
|
| 22 |
+
print("cuda available:", torch.cuda.is_available())
|
| 23 |
+
print("cuda device count:", torch.cuda.device_count())
|
| 24 |
+
if torch.cuda.is_available():
|
| 25 |
+
print("current device:", torch.cuda.current_device())
|
| 26 |
+
print("device name:", torch.cuda.get_device_name(torch.cuda.current_device()))
|
| 27 |
+
|
| 28 |
print("Using device:", device)
|
| 29 |
|
| 30 |
from diffusers import FlowMatchEulerDiscreteScheduler
|