Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -80,6 +80,7 @@ md = "![](https://drive.google.com/uc?id=1WBqYsKRwn_78A72MJBrk643l3-gfAssP) \n"
|
|
80 |
"This project is licensed under the terms of the MIT license. \n"
|
81 |
|
82 |
if __name__ =='__main__':
|
|
|
83 |
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
84 |
# True
|
85 |
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
|
|
80 |
"This project is licensed under the terms of the MIT license. \n"
|
81 |
|
82 |
if __name__ =='__main__':
|
83 |
+
torch.cuda.init()
|
84 |
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
85 |
# True
|
86 |
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|