查看CUDA、Python和PyTorch版本兼容性顺序及相关网站推荐

任务:

在一个有GPU的电脑上安装pytorch时选择合适版本。

简述:

安装 pytorch 时,若要使用 GPU,需要考虑与 CUDA,python 的版本兼容情况。推荐考虑顺序:CUDA 电脑GPU可/在使用的版本,pytorch可兼容CUDA的版本,python可兼容 pytorch的版本。

  • 注意,版本大多向下兼容。
  • 细节描述:

    1. 查询版本兼容的顺序:

    2. 先查看电脑的GPU安装的cuda版本。
    3. 右键电脑桌面的空白处,单击"NVIDIA控制面板",点击其左下角"系统信息",选择弹出窗口内的"组件",即可查看 CUDA 版本信息。
    4. 再查看该cuda版本可兼容的 pytorch 版本有哪些。对应官网。
    5. 然后查看和那些 pytorch 版本兼容的 python 版本们,依此选择合适的 python版本。对应官网。
    6. 接着,安装python,pytorch及其依赖项。

    7. 先创建虚拟环境,安装选好的 python 版本。
    8. 安装和删除环境命令:
    9. conda create -n env_name python=XX # 所选python版本
    10. conda remove -n env_name -y –all # -y 直接卸载,省去按 y,可加可不加。
    11. 再依据官网命令,安装对应的 pytorch 版本,命令中一般应该会包含cuda相关的内容。
    物联沃分享整理
    物联沃-IOTWORD物联网 » 查看CUDA、Python和PyTorch版本兼容性顺序及相关网站推荐

    发表评论