Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
Signed-off-by: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com>
  • Loading branch information
KernFerm authored Oct 17, 2024
1 parent 5222b62 commit 42dd3b7
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions get_device.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
import torch

print(torch.cuda.is_available()) # This will return True if CUDA is available
print(torch.version.cuda) # This will print the CUDA version being used
print(torch.cuda.get_device_name(0)) # This will print the name of the GPU, e.g., 'NVIDIA GeForce RTX GPU Model'

0 comments on commit 42dd3b7

Please sign in to comment.