Skip to content

Instantly share code, notes, and snippets.

@KushajveerSingh

KushajveerSingh/post_0017_13

Created October 7, 2021 03:55
Show Gist options
  • Save KushajveerSingh/64b2d83e4ef1d5a1462ec1cf20821d4b to your computer and use it in GitHub Desktop.
Save KushajveerSingh/64b2d83e4ef1d5a1462ec1cf20821d4b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
post_0017_13
(base) default@7d9b75595a27:~$ python
Python 3.9.7 (default, Sep 16 2021, 13:09:58)
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>> torch.backends.cudnn.version()
8005
>>> x = torch.tensor([1,2], device='cuda:0')
>>> x
tensor([1, 2], device='cuda:0')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment