PyTorch Cuda 与 anaconda 不可用答案

【问题标题】：PyTorch Cuda with anaconda not availablePyTorch Cuda 与 anaconda 不可用
【发布时间】：2019-04-05 17:53:51
【问题描述】：

我正在使用 anaconda 来调节我的环境，对于一个项目，我必须使用我的 GPU 进行网络训练。我在我的项目中使用 pytorch，我正在尝试让 CUDA 正常工作。

我安装了 cudatoolkit、numba、cudnn

仍然，当我尝试这个命令时：

torch.cuda.is_available()

我得到“False”作为输出。这是我的环境：

# Name                    Version                   Build  Channel
blas                      1.0                         mkl  
bzip2                     1.0.6                h470a237_2    conda-forge
ca-certificates           2018.03.07                    0  
cairo                     1.14.12              he6fea26_5    conda-forge
certifi                   2018.8.24                py35_1  
cffi                      1.11.5           py35he75722e_1  
cloudpickle               0.5.5                    py35_0  
cudatoolkit               9.2                           0    anaconda
cudnn                     7.2.1                 cuda9.2_0    anaconda
cycler                    0.10.0                     py_1    conda-forge
cython                    0.28.5           py35hf484d3e_0    anaconda
dask-core                 0.19.2                   py35_0  
dbus                      1.13.0               h3a4f0e9_0    conda-forge
decorator                 4.3.0                    py35_0  
expat                     2.2.5                hfc679d8_2    conda-forge
ffmpeg                    4.0.2                ha0c5888_1    conda-forge
fontconfig                2.13.1               h65d0f4c_0    conda-forge
freetype                  2.9.1                h6debe1e_4    conda-forge
gettext                   0.19.8.1             h5e8e0c9_1    conda-forge
giflib                    5.1.4                h470a237_1    conda-forge
glib                      2.55.0               h464dc38_2    conda-forge
gmp                       6.1.2                hfc679d8_0    conda-forge
gnutls                    3.5.19               h2a4e5f8_1    conda-forge
graphite2                 1.3.12               hfc679d8_1    conda-forge
gst-plugins-base          1.12.5               hde13a9d_0    conda-forge
gstreamer                 1.12.5               h61a6719_0    conda-forge
harfbuzz                  1.9.0                h08d66d9_0    conda-forge
hdf5                      1.10.2               hc401514_2    conda-forge
icu                       58.2                 hfc679d8_0    conda-forge
imageio                   2.4.1                    py35_0  
intel-openmp              2019.0                      118  
jasper                    1.900.1              hff1ad4c_5    conda-forge
jpeg                      9c                   h470a237_1    conda-forge
kiwisolver                1.0.1            py35h2d50403_2    conda-forge
libedit                   3.1.20170329         h6b74fdf_2  
libffi                    3.2.1                hd88cf55_4  
libgcc-ng                 8.2.0                hdf63c60_1  
libgfortran               3.0.0                         1    conda-forge
libgfortran-ng            7.3.0                hdf63c60_0  
libiconv                  1.15                 h470a237_3    conda-forge
libopenblas               0.3.3                h5a2b251_3  
libpng                    1.6.35               ha92aebf_2    conda-forge
libstdcxx-ng              8.2.0                hdf63c60_1  
libtiff                   4.0.9                he6b73bb_2    conda-forge
libuuid                   2.32.1               h470a237_2    conda-forge
libwebp                   0.5.2                         7    conda-forge
libxcb                    1.13                 h470a237_2    conda-forge
libxml2                   2.9.8                h422b904_5    conda-forge
llvmlite                  0.24.0           py35hdbcaa40_0  
matplotlib                3.0.0            py35h0b34cb6_1    conda-forge
mkl                       2019.0                      118  
mkl_fft                   1.0.6                    py35_0    conda-forge
mkl_random                1.0.1                    py35_0    conda-forge
ncurses                   6.1                  hf484d3e_0  
nettle                    3.3                           0    conda-forge
networkx                  2.1                      py35_0  
ninja                     1.8.2            py35h6bb024c_1  
numba                     0.39.0           py35h04863e7_0  
numpy                     1.15.2           py35h1d66e8a_0  
numpy-base                1.15.2           py35h81de0dd_0  
olefile                   0.46                     py35_0  
openblas                  0.2.20                        8    conda-forge
opencv                    3.4.1            py35h6fd60c2_1  
opencv-python             3.4.3.18                  <pip>
openh264                  1.7.0                         0    conda-forge
openssl                   1.0.2p               h14c3975_0  
pandas                    0.23.4           py35h04863e7_0  
pcre                      8.41                 hfc679d8_3    conda-forge
pillow                    5.2.0            py35heded4f4_0  
Pillow                    5.3.0                     <pip>
pip                       10.0.1                   py35_0  
pixman                    0.34.0               h470a237_3    conda-forge
pthread-stubs             0.4                  h470a237_1    conda-forge
pycparser                 2.19                     py35_0  
pyparsing                 2.2.2                      py_0    conda-forge
pyqt                      5.6.0            py35h8210e8a_7    conda-forge
python                    3.5.6                hc3d631a_0  
python-dateutil           2.7.3                      py_0    conda-forge
pytorch                   0.4.1           py35_py27__9.0.176_7.1.2_2    pytorch
pytz                      2018.5                   py35_0  
pywavelets                1.0.0            py35hdd07704_0  
qt                        5.6.2                hf70d934_9    conda-forge
readline                  7.0                  h7b6447c_5  
scikit-image              0.14.0           py35hf484d3e_1  
scipy                     1.1.0            py35hfa4b5c9_1  
setuptools                40.2.0                   py35_0  
sip                       4.18.1           py35hfc679d8_0    conda-forge
six                       1.11.0                   py35_1    conda-forge
sqlite                    3.25.2               h7b6447c_0  
tk                        8.6.8                hbc83047_0  
toolz                     0.9.0                    py35_0  
torchvision               0.1.9            py35h72e4c6f_1    soumith
tornado                   5.1.1            py35h470a237_0    conda-forge
wheel                     0.31.1                   py35_0  
x264                      1!152.20180717       h470a237_1    conda-forge
xorg-kbproto              1.0.7                h470a237_2    conda-forge
xorg-libice               1.0.9                h470a237_4    conda-forge
xorg-libsm                1.2.2                h8c8a85c_6    conda-forge
xorg-libx11               1.6.6                h470a237_0    conda-forge
xorg-libxau               1.0.8                h470a237_6    conda-forge
xorg-libxdmcp             1.1.2                h470a237_7    conda-forge
xorg-libxext              1.3.3                h470a237_4    conda-forge
xorg-libxrender           0.9.10               h470a237_2    conda-forge
xorg-renderproto          0.11.1               h470a237_2    conda-forge
xorg-xextproto            7.3.0                h470a237_2    conda-forge
xorg-xproto               7.0.31               h470a237_7    conda-forge
xz                        5.2.4                h14c3975_4  
zlib                      1.2.11               ha838bed_2

我的桌面有一个 NVIDIA GeForce GTX 970（所以它是 cuda 可用的）同样出于某种原因，您可以在此处看到：

我的显卡不显示，但是在使用时

lspci -v

命令，我可以在那里看到我的显卡。不知道这是否与它有关。任何人都知道我该如何解决这个问题？

【问题讨论】：

标签： anaconda conda pytorch pycuda

【解决方案1】：

您需要使用https://pytorch.org/get-started/locally/“一次性”安装 pytorch 以构建 anaconda 命令。

使用anaconda中的标准配置，你会得到：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

（请始终检查https://pytorch.org/get-started/locally/此命令是否仍然是最新的。）

您似乎需要正确的 cuda 10.2 版软件包才能与 pytorch 可以处理的内容保持一致。这就是@RussellGallop 的有用信息的含义。

我们可以看到不建议单独安装 pytorch 和 cuda，建议安装 Anaconda，与您的回答相反：

Anaconda 是我们推荐的包管理器，因为它会安装所有依赖项。

卸载安装比修复好

如果出现问题，您最好卸载所有涵盖的软件包并应用https://pytorch.org/get-started/locally/ 以再次获取命令，而不是尝试通过单独安装来修复它。因此，如果您想卸载，您需要使用与安装完全相同的命令，但要使用“conda uninstall”。

对于上面的例子，卸载命令是：

conda uninstall pytorch torchvision cudatoolkit=10.2 -c pytorch

再次需要“一次性”卸载是对 pytorch 敏感安装的另一个提示，并且单独安装是有风险的。见How can l uninstall PyTorch?)

【讨论】：

特别注意-c pytorch。 conda-forge 等其他渠道可能无法使用。
如果你使用-c conda-forge，你可能想要conda install pytorch-gpu

【解决方案2】：

固定：我单独安装了 CUDA，而不是通过 anaconda，现在它可以工作了。如果有人知道为什么在anaconda中安装cuda时不起作用，请随时回答

【讨论】：

不推荐，最好使用conda tensorflow或pytorch安装器的依赖自动安装。

【解决方案3】：

您的 CUDA 安装需要与您的 NVidia 驱动程序（不是由 Anaconda 安装的）兼容。我在 Windows 上与 Anaconda 有类似的情况。我忽略了 torch.cuda.is_available() 检查并尝试将模型发送到 GPU：

device = torch.device("cuda")
model.to(device)

这给出了更有用的信息：

The NVIDIA driver on your system is too old (found version 10010).
Please update your GPU driver by downloading and installing a new
version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install
a PyTorch version that has been compiled with your version
of the CUDA driver.

【讨论】：