Spaces:
Runtime error
Runtime error
measmonysuon
commited on
Commit
•
ca60ac1
1
Parent(s):
cd028fb
Update app.py
Browse files
app.py
CHANGED
@@ -30,6 +30,7 @@ client_image = Client("measmonysuon/DALLE-4K")
|
|
30 |
user_info = whoami(token=API_TOKEN)
|
31 |
print("User Info:", user_info)
|
32 |
|
|
|
33 |
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
34 |
# True
|
35 |
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
|
|
30 |
user_info = whoami(token=API_TOKEN)
|
31 |
print("User Info:", user_info)
|
32 |
|
33 |
+
print(torch.version.cuda)
|
34 |
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
35 |
# True
|
36 |
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|