1. 首先打开终端
查看目前的环境
conda env list
可以看到,目前我只有base环境,如果有其他虚拟环境也会显示出来
2. 创建虚拟环境----conda create -n [环境名] python=3.7.5
conda create -n JuliaYeung python=3.7.5
3. 输入y
4. 安装完成
5. 激活虚拟环境
source activate JuliaYeung
可以看到前方括号出现虚拟环境名称
6. 安装pytorch
6.1 查看cuda 版本
6.1.1 重开一个终端或者退出当前anaconda虚拟环境
source deactivate JuliaYeung
6.1.2 nvidia-smi 查看cuda版本
nvidia-smi
可以看到我的是cuda版本是11.1
6.2 进入pytorch官网
根据你的情况选择,复制最底行命令进行安装
图中可以看到,只有cuda10.2, cuda11.3, 和cuda11.6 的选择
我的cuda版本为11.1
点击 install previous versions of PyTorch
往下翻,选择你想要的PyTorch 版本和 你适用的cuda版本
我选择PyTorch1.10.1和cuda11.1的命令
复制到终端
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
等待安装
安装完成!!
开启快乐学习。