cuda、python、pytorch 版本兼容查看顺序及网站

任务：

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

简述：

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

• 注意，版本大多向下兼容。

细节描述：

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

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

2. 接着，安装python，pytorch及其依赖项。

   • 先创建虚拟环境，安装选好的 python 版本。
     • 安装和删除环境命令：
       • conda create -n env_name python=XX # 所选python版本
       • conda remove -n env_name -y --all # -y 直接卸载，省去按 y，可加可不加。
   • 再依据官网命令，安装对应的 pytorch 版本，命令中一般应该会包含cuda相关的内容。