任务:
在一个有GPU的电脑上安装pytorch时选择合适版本。
简述:
安装 pytorch 时,若要使用 GPU,需要考虑与 CUDA,python 的版本兼容情况。推荐考虑顺序:CUDA 电脑GPU可/在使用的版本,pytorch可兼容CUDA的版本,python可兼容 pytorch的版本。
- 注意,版本大多向下兼容。
细节描述:
-
查询版本兼容的顺序:
-
接着,安装python,pytorch及其依赖项。
- 先创建虚拟环境,安装选好的 python 版本。
- 安装和删除环境命令:
- conda create -n env_name python=XX # 所选python版本
- conda remove -n env_name -y --all # -y 直接卸载,省去按 y,可加可不加。
- 安装和删除环境命令:
- 再依据官网命令,安装对应的 pytorch 版本,命令中一般应该会包含cuda相关的内容。
- 先创建虚拟环境,安装选好的 python 版本。