Я установил PyTorch, но CUDA не распознается

Question

Я установил PyTorch с помощью этой команды:

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

И, похоже, это сработало.При импорте факела не возникает никаких ошибок.Мой ноутбук имеет GeForce 1060 TI , который, как я предполагал, будет работать с CUDA.

Вот ошибка в IDE (Eclipse):

Answer 1

Вы должны включить код, а не график!

Удалите PyTorch и установите его снова с помощью этой команды:

conda install pytorch -c pytorch
pip3 install torchvision

после этого попробуйте проверить его с помощью этогокоманда:

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 ...')