Skip to content

Instantly share code, notes, and snippets.

@CaptainAshis

CaptainAshis/check_cuda.py

Last active August 25, 2018 04:56
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save CaptainAshis/32ccd4ab4914338feca542fb7a0391ce to your computer and use it in GitHub Desktop.
Save CaptainAshis/32ccd4ab4914338feca542fb7a0391ce to your computer and use it in GitHub Desktop.
Download ZIP
cuda fastai
Raw
check_cuda.py
torch.cuda.is_available()
#True
torch.backends.cudnn.enabled
#True
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment