Created
January 22, 2020 10:20
-
-
Save luminoso/f331c4f5be1258647d134f004aa56634 to your computer and use it in GitHub Desktop.
test CUDA avaialbility pytorch
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# tests the avalability of cuda for pytorch | |
import torch | |
torch.cuda.current_device() | |
torch.cuda.device(0) | |
torch.cuda.device_count() | |
torch.cuda.get_device_name(0) | |
torch.cuda.is_available() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment