任务:
在一个有GPU的电脑上安装pytorch时选择合适版本。
简述:
安装 pytorch 时,若要使用 GPU,需要考虑与 CUDA,python 的版本兼容情况。推荐考虑顺序:CUDA 电脑GPU可/在使用的版本,pytorch可兼容CUDA的版本,python可兼容 pytorch的版本。
细节描述:
-
查询版本兼容的顺序:
- 先查看电脑的GPU安装的cuda版本。
- 右键电脑桌面的空白处,单击"NVIDIA控制面板",点击其左下角"系统信息",选择弹出窗口内的"组件",即可查看 CUDA 版本信息。
- 再查看该cuda版本可兼容的 pytorch 版本有哪些。对应官网。
- 然后查看和那些 pytorch 版本兼容的 python 版本们,依此选择合适的 python版本。对应官网。
-
接着,安装python,pytorch及其依赖项。
- 先创建虚拟环境,安装选好的 python 版本。
- 安装和删除环境命令:
- conda create -n env_name python=XX # 所选python版本
- conda remove -n env_name -y --all # -y 直接卸载,省去按 y,可加可不加。
- 再依据官网命令,安装对应的 pytorch 版本,命令中一般应该会包含cuda相关的内容。