打开Anaconda Prompt
在命令行格式下,输入代码,完成调用清华镜像、建立pytorch环境、安装pytorch、测试pytorch过程
使用清华镜像
分别输入以下四句代码
1 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
2 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
3 conda config --set show_channel_urls yes
4 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
创建Pytorch环境
(1)创建环境pytorch,使用Python版本是3.7(这一步很重要,直接使用默认环境可能由于python版本问题无法安装pytorch)
1 conda create -n pytorch python=3.7
之后,弹出提示,输入 y,即可安装。
(2)查看环境是否安装成功
1 conda info --envs
可以看到base和pytorch两个环境
切换到pytorch环境:activate pytorch
安装pytorch
根据自己的安装版本,在pytorch网站寻找代码
网站:PyTorch
使用conda安装失败,于是用pip安装成功(为了简单安装,使用了cpu版本)
复制代码(以下代码在python3.5好像会失败):
pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
安装ipykernel(为了在jupyter notebook中可以选择环境)
pip install ipykernel -i https://pypi.douban.com/simple
将虚拟环境引用到Jupyter Notebook中(pytorch为自己的环境名,可以更改)
python -m ipykernel install --user --name pytorch --display-name "Python [conda env:pytorch]"
成功后,打开jupyter notebook 如图
最后测试python版本与pytorch
import torch
import sys
sys.version