我已经安装了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 ...')