我已经安装了 PyTorch，但 CUDA 无法识别

【问题标题】：I've installed PyTorch, but CUDA is not recognized我已经安装了 PyTorch，但 CUDA 无法识别
【发布时间】：2020-01-20 00:00:53
【问题描述】：

我使用这个命令安装了PyTorch：

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

而且它似乎奏效了。导入torch时，没有任何错误。我的笔记本电脑有一个 GeForce 1060 TI，我认为它可以与 CUDA 一起使用。

这是 IDE (Eclipse) 中的错误：

【问题讨论】：

Stack Overflow Discourages screenshots of code, errors & data

标签： python pytorch

【解决方案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 ...')

【讨论】：