pytorch 支持amd显卡吗_为什么PyTorch找不到我的NVDIA驱动程序来支持CUDA？

我在我的机器中添加了一个geforcegtx1080ti(运行ubuntu18.04和Anaconda with python3.7)，以便在使用PyTorch时使用GPU。两张卡都正确识别：$ lspci | grep VGA

03:00.0 VGA compatible controller: NVIDIA Corporation GF119 [NVS 310] (reva1)

04:00.0 VGA compatible controller: NVIDIA Corporation GP102 [GeForce GTX 1080 Ti] (rev a1)

NVS 310处理我的2显示器设置，我只想利用1080的Pythorch。我还安装了最新的NVIDIA驱动程序，这些驱动程序目前在存储库中，而且看起来还不错：

^{pr2}$

驱动程序版本390.xx允许根据NVIDIA docs运行CUDA9.1(9.1.85)。简单的CUDA版本也安装在Ubuntu工具箱中：$ sudo apt-get-installed nvidia-cuda-toolkit

再说一次，这似乎没问题：$ nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2017 NVIDIA Corporation

Built on Fri_Nov__3_21:07:56_CDT_2017

Cuda compilation tools, release 9.1, V9.1.85

以及$ apt-cache policy nvidia-cuda-toolkit

nvidia-cuda-toolkit:

Installed: 9.1.85-3ubuntu1

Candidat