提示
pytorch的安装经过了几次变化,请大家以官网的安装命令为准。另外需要说明的就是在1.2版本以后,pytorch只支持cuda 9.2以上了,所以需要对cuda进行升级,目前测试大部分显卡都可以用,包括笔记本的MX250也是可以顺利升级到cuda 10.1。
安装根据官网,对系统选择和安装方式等灵活选择即可。 为提高下载速度,在国内安装anaconda建议使用清华镜像。
我选择的版本为
?2.安装
默认选项即可。
?3.环境配置
在电脑设置中找到高级系统设置,选择环境变量。
?选择Path,点击编辑。
?根据自己的安装路径添加
?环境配置完成后,打开Anaconda Navigator
1.官网安装
pytorch的安装经过了几次变化,以官网的安装命令为准。另外需要说明的就是在1.2版本以后,pytorch只支持cuda 9.2以上了,所以需要对cuda进行升级,目前测试大部分显卡都可以用,包括笔记本的MX250也是可以顺利升级到cuda 10.1。
建议大家使用pip进行安装。
得到相应的命令:
#默认 使用 cuda10.2
pip3 install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.htm
#cuda 11.3
pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
#cpu版本
pip3 install torch torchvision torchaudio
2.在cmd运行给出的命令
命令:
pip3 install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.htm
?
总结
根据以上就安装完成啦。
|