1.安装conda(Miniconda、Anaconda都可以)
以下默认的是python3.8,pytorch选择的是CPU版本!实际可以根据需求安装任意版本。如果CUDA不匹配的可以卸载原来的CUDA安装新的CUDA。
2.GPU版本需要确定CUDA版本
在桌面点击右键,选择“NVIDIA控制面板”
依次点击:帮助-->系统信息
点击“组件”,如图所示,本电脑CUDA是11.4的
3.下载Pytorch的离线包
推荐网站:https://download.pytorch.org/whl/torch_stable.html
技巧:可以打开pytorch的官网PyTorch查看自己适合什么版本(注意选择CUDA版本!),然后再进入上面的网站单独下载
如:
对应的分别下载(%2B应该是beta版本,实际可以下载正式版本):
最后下载得到的有:
3.安装
- 首先打开cmd
- 使用“conda activate xxxxx(之前创建的环境名字)”激活到当前环境
- 再使用cd定位到下载torch的目录中
- 最后依次pip install torch、torchvison、torchaudio(一定要输入完整的名字)
- 全部安装完了试一试输入python进入解释器,再输入import torch。如果没有报错即安装成功。
- 附:安装过程中可能有些包的下载速度有些慢,可以使用清华源替换.condarc这个文件,详细参考清华源的界面介绍