evanslight / caffe-gpu-installation Goto Github PK

sudo apt-get install -y build-essential cmake git pkg-config

sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler

sudo apt-get install -y libatlas-base-dev

sudo apt-get install -y --no-install-recommends libboost-all-dev

sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev

sudo apt-get install -y python-pip

sudo apt-get install -y python-dev

sudo apt-get install -y python-numpy python-scipy

sudo apt-get install -y libopencv-dev

#for cuda.deb (I got Error Unmet dependencies for last vesrion of cuda 8.0.61 so better download .run file):
#sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
#sudo apt-get update
#sudo apt-get install cuda
#sudo apt-get install cuda-drivers

#for cuda.run file (follow the on-screen prompts):

sudo sh cuda_8.0.61_375.26_linux.run

#next for both

echo 'export PATH=/usr/local/cuda-8.0/bin:$PATH' » ~/.bashrc

echo 'export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc

#restart terminal

cd /usr/local/cuda-8.0/samples/

sudo make all -j8

#create folder in home directory. for example

mkdir ~/code

cd ~/code

tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz #or extract cudnn file

cd cuda/lib64/

sudo cp lib* /usr/local/cuda-8.0/lib64/

cd cuda/include/

sudo cp cudnn.h /usr/local/cuda-8.0/include/

cd code

git clone https://github.com/BVLC/caffe.git

cd caffe/python/

for req in $(cat requirements.txt); do sudo -H pip install $req --upgrade; done

sudo apt-get update

#export PYTHONPATH to your python folder in caffe folder (don't forget to write your user name and folder correctly)
#for example:

echo 'export PYTHONPATH=/home/user/code/caffe/python:$PYTHONPATH' >> ~/.bashrc


cd /usr/lib/x86_64-linux-gnu

sudo ln -s libhdf5_serial.so.10.1.0 libhdf5.so

sudo ln -s libhdf5_serial.so.10.1.0 libhdf5_hl.so


cd caffe

cp Makefile.config.example Makefile.config


#check make.config file in caffe folder:

cd caffe/build

cmake ..

cd code/caffe

make all -j8

make test

sudo reboot

make runtest

cd code/caffe

make pycaffe

make distribute

python

import caffe