服务器Ubuntu、pytorch框架、网络模型SE-Resnet50,优化算法Adam
pytorch(python优先的深度学习框架,是一个和tensorflow,Caffe,MXnet一样,非常底层的框架,它的前身是torch)
主要步骤(环境配置):
1. 需要下载cuda8.0
2. 需要下载pytorch(两种方法)
- 第一种方式:可以下载Anaconda(一个python的包管理工具),通过conda下载pytorch
- 第二种方式:可以直接下载pytorch(如以下命令行)
pip install torchvision
pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp27-cp27m-linux_x86_64.whl
第一种方式(下载的Anaconda,虽然比较大,但也可以用于其他python应用,十分方便)
1. 安装cuda8.0/cuda9.0
自行搜索教程(无坑)
2. 安装Anaconda
wegt https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda2-5.1.0-Linux-x86_64.sh
cd download
ls
bash Anaconda2-5.1.0-Linux-x86_64.sh
# 然后一路enter,最后yes
#配置conda环境变量(一直按enter键,到最后有一个让添加环境变量,你写yes,如果没有就需要手动添加)
http://www.cnblogs.com/gaofighting/p/8799169.html通过echo命令行手动添加环境变量
# 使用conda list查看安装的包,测试是否正确安装
# 置换镜像
conda 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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
3. 安装pytorch
conda install pytorch torchvision -c pytorch
注意:如果机子上安装的cuda是9.0:conda install pytorch torchvision cuda90 -c pytorch
4. 验证pytorch是否正确安装
python
import torch