ubuntu下安装person_reid环境并训练

环境:Ubuntu 14.04

下载person_reid:https://github.com/layumi/Person_reID_baseline_pytorch

1.首先需要安装python3.6环境和numpy,因为Ubuntu14.04默认安装为python2.7,在此需要借助pyenv进行python版本管理

以下为安装python3.6&&pip3&&numpy

$ git clone git://github.com/yyuu/pyenv.git ~/.pyenv
$ echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bashrc
$ echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bashrc
$ echo 'eval "$(pyenv init -)"' >> ~/.bashrc
$ exec $SHELL -l

$ sudo apt-get install libc6-dev gcc

$ sudo apt-get install -y make build-essential libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm

安装python:
$ pyenv install 3.6.1 -v

更新数据库:
$ pyenv rehash

查看当前已经安装的python版本
$ pyenv versions
* system (set by /home/seisman/.pyenv/version)
3.6.1
星号代表是当前系统正在使用的python版本

设置全局的python版本
$ pyenv global 3.6.1
$ pyenv versions
system
* 3.6.1 (set by /home/seisman/.pyenv/version)

安装pip3:
$ sudo apt-get install python3-pip
pyenv rehash
$ pip3 install numpy

$ pip3 install scipy

2.安装cuda7.5和cudnn

ctrl+alt+F1(tty1)(如果进入tty1失败可以编辑/etc/default/grub修改GRUB_CMDLINE_LINUX_DEFAULT的值为nomodeset更新grub:sudo update-grub重启)
输入用户名密码
sudo service lightdm stop(如果下面程序安装出错重新进入tty1需再次执行该命令)
chmod +x ./cuda_7.5_linux_run
./cuda_7.5_linux_run
REMEMBER: Do not install openGL
sudo reboot 
sudo apt-get install libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler libboost-all-dev
sudo apt-get update
extracting cudnn-7.0-linux-x64-v3.0-rc.tgz
cd cuda/lib64
sudo cp lib* /usr/local/cuda-7.5/lib64/
cd ../include
sudo cp cudnn.h /usr/local/cuda-7.5/include/
sudo ldconfig

sudo reboot

3.安装Pytorch 0.3.0 and Torchvision 0.2.0

进入http://pytorch.org/下载基于cuda7.5和python3.6的pytorch并通过以下命令安装

pip install torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl

进入https://github.com/pytorch/vision下载0.2.0版vision(目前通过git clone https://github.com/pytorch/vision下载的vision为0.2.1版本,训练时会报错,只能手动找到0.2.0版本)

cd vision//进入vision目录

python setup.py install//安装vision

4.下载Market1501 Datasethttp://www.liangzheng.org/Project/project_reid.html

cd Person_reID_baseline_pytorch-master//进入person_reid目录

python prepare.py//修改文件内path,进行训练图片准备阶段

训练命令:python train.py --gpu_ids 0 --name ft_ResNet50 --train_all --batchsize 32  --data_dir /home/hyzn/Market-1501-v15.09.15/pytorch






你可能感兴趣的:(机器学习)