SyntaxFix
Write A Post
Hire A Developer
Questions
Create a tensor on the GPU as follows:
$ python >>> import torch >>> print(torch.rand(3,3).cuda())
Do not quit, open another terminal and check if the python process is using the GPU using:
$ nvidia-smi