深度学习--第2篇: Anaconda+Pytorch(CPU版本)环境配置

1. Anaconda下载安装

参考上一篇博客: 博客链接

2. 安装Pytorch

2.1 创建虚拟环境

conda create -n pytorch python=3.6

2.2 激活虚拟环境

# 激活
source activate pytorch
# 关闭
source deactivate pytorch

2.3 安装CPU版本

# 官网安装CPU版本,网速很慢
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch
# 清华镜像安装CPU版本,网速很快
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly

# 官网安装GPU版本,很慢
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
# 清华镜像安装GPU版本,网速很快
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0

conda install cudatoolkit=10.1

2.4 测试安装

• 运行以下命令,如果不报错,表示安装成功

source activate pytorch
python
import torch
import torchvision

2.5 安装模型可视化工具

安装torchsummart实现模型输出可视化

cd ~/anaconda3/envs/pytorch/lib/python3.6/site-packages
git clone https://github.com/sksq96/pytorch-summary
cd pytorch-summary
cp -r torchsummary/ ../torchsummary
再删除pytorch-summary文件夹即可

# 导入功能函数即可使用
from torchsummary import summary