刚刚开始学习,由于我的笔记本没有独立显卡,所以先尝试着装一个CPU版本的。
环境准备
anaconda3(官网直接下载安装即可)创建conda虚拟环境
考虑到以后可能需要使用到多个环境,随意在conda中创建一个虚拟环境来支持pytorch,打开anaconda promptconda create -n pytorch(虚拟环境名字) python=3.7 1
安装pytorch
安装pytorch,直接从官网下载会很慢,所以我们可以配置镜像源,如清华源镜像的配置:onda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 1234
配置完成后我们可以去官网获取安装命令,如下图。
注意:命令中去掉最后的-c pytorch,否则还是从官方源下载,速度很慢,且容易失败。
执行安装等到安装完成即可。
将下载的压缩文件存放的你想存放的文件夹内,在命令行中,进入该目录下。执行安装
conda install -n pytorch pytorch-1.5.1-py3.7_cpu_0.tar.bz2 conda install -n pytorch torchvision-0.6.1-py37_cpu.tar.bz2 12 我们来测试一下,首先激活虚拟环境
activate pytorch python 12 在python中输入
import torch import torchvision 12
没有错误提示说明安装成功。
Jupyter notebook配置
如果要在juptyer notebook中使用该虚拟环境,首先我们需要安装下ipykernelconda install -n pytorch(虚拟环境名字) ipykernel 1 将环境写入notebook的kernel中。
python -m ipykernel install --user --name 环境名称 --display-name "在jupyter中显示的环境名称" 1 启动jupyter notebook
jupyter notebook 1
自动跳转后,我们点击右上角New,发现我们新建立的环境pytorch已经检测到了。如下图。
我们新建一个pytorch文件,导入包文件,未报错,大功告成。