Pytorch cuda 版本无法安装答案

【问题标题】：Pytorch cuda version can not installPytorch cuda 版本无法安装
【发布时间】：2021-02-27 15:43:10
【问题描述】：

一开始，我使用nvcc --version 命令检查了我的 cuda 版本，它显示版本为 10.2 所以我开始根据pytorch 中的说明使用 cuda 安装 pytorch，所以我尝试在 python 3.8.3 虚拟环境的 anaconda 提示符下使用波纹管命令

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

在终端显示

    added / updated specs:
        - cudatoolkit=10.2
        - pytorch
        - torchaudio
        - torchvision

The following NEW packages will be INSTALLED:

    cudatoolkit: 10.2.89-h74a9793_1
    pytorch:     1.7.0-py3.8_cpu_0  pytorch [cpuonly]
    torchaudio:  0.7.0-py38         pytorch
    torchvision: 0.8.1-py38_cpu     pytorch [cpuonly]

安装过程中没有任何错误，然后我通过激活此安装环境尝试在笔记本中使用以下代码

import torch
print(torch.__version__, torch.cuda.get_device_properties(0) if torch.cuda.is_available() else 'CPU')

输出为1.7.0 CPU

所以我在 anaconda 提示符下运行conda list PyTorch 命令

它还显示在输出下方

# Name                    Version                   Build  Channel
pytorch                   1.7.0               py3.8_cpu_0  [cpuonly]  pytorch

为什么它没有安装 GPU 版本，我该如何解决这个问题

谢谢

【问题讨论】：

你有什么类型的 GPU？您的 GPU 驱动程序是什么版本？
@jodag GPU = nvidia 940MX 和驱动程序版本 = 456.38
糟糕删除了之前的评论，因为它不正确。问题是，虽然您的卡支持 cuda 10.2，但 pytorch 二进制文件并未提供对计算能力 5.0 的支持。这意味着您需要从源代码重建 pytorch。
@jodag 能否请您提供任何文档资源链接或相关指南
如果您使用的是 Windows 10，this answer 也适用

标签： python pytorch gpu

【解决方案1】：

我正在使用 Python 版本 3.6.13 和你在这里使用的一样。您可以尝试在另一个虚拟环境中更改 Python 版本，它应该可以工作。

PS：要记住的是我的操作系统是 Windows 10

PyTorch GPU 工作的图片：

【讨论】：