1. 安装Anaconda
1.1 版本
版本无所谓,安装个稳定的就可以,博主在旧版本上安装库时,没有出现兼容问题。
1.2 配置环境变量
安装好后要配置环境变量
2. 安装Pytorch
2.1 创建虚拟环境
2.2 电脑本地安装CUDA(只支持Nvidia显卡)
这里要注意:电脑上安装的是CUDA,Anaconda虚拟环境中安装torch时用到的cudatoolkit是CUDA的一个子集,我们要保证CUDA版本>cudatoolkit的版本。
2.3 配置CUDA的环境变量
24. 安装Pytorch
用pip安装
其实是安装这三个库:torch、torchvision、torchaudio
要注意这三个库的版本,要与python对应上