I've installed PyTorch, but CUDA is not recognized
I installed PyTorch using this command:

pip3 install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html


And it seemed to work. When importing torch, there isn't any error. 
My laptop has a GeForce 1060 TI, which I assumed would work with CUDA.

Here is the error in the IDE (Eclipse):



You should include code, not a graph!

Remove PyTorch and install it again with this command:

conda install pytorch -c pytorch
pip3 install torchvision


after this try checking it with this command:

import torch
import torchvision
train_on_gpu = torch.cuda.is_available()

if train_on_gpu:
    print('CUDA is available, Training on GPU ...')
else:
    print('CUDA is not available!  Training on CPU ...')

