Friday, December 16, 2016

Ubuntu 16.04 + GTX1080 + Caffe

USB install Ubuntu 16.04

sudo apt-get install ssh
sudo service ssh restart

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-367 (no need to upgrade to 370 as CUDA 8 use 367)

download 

linux-headers-4.7.0-040700_4.7.0-040700.201608021801_all.deb

linux-headers-4.7.0-040700-generic_4.7.0-040700.201608021801_amd64.deb

linux-image-4.7.0-040700-generic_4.7.0-040700.201608021801_amd64.deb




sudo dpkg -i *.deb



upgrade to kernel 4.7

fix Intel Dual BandWireless-AC 3168



sudo dpkg -i linux-firmware_1.161_all.deb


CUDA

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.44-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda


https://yangcha.github.io/GTX-1080/ 
cuDNN 5.1
tar xvzf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
 


Install Caffe
https://github.com/BVLC/caffe/wiki/Ubuntu-16.04-or-15.10-Installation-Guide 

sudo apt-get update

sudo apt-get -y upgrade

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

# (Python general)
sudo apt-get install -y python-pip

# (Python 2.7 development files)
sudo apt-get install -y python-dev
sudo apt-get install -y python-numpy python-scipy

# (or, Python 3.5 development files)
sudo apt-get install -y python3-dev
sudo apt-get install -y python3-numpy python3-scipy

# (OpenCV 2.4)
sudo apt-get install -y libopencv-dev
 
git clone https://github.com/BVLC/caffe.git

cd caffe
 
cp Makefile.config.example Makefile.config

vi Makefile.config
 
cd python

for req in $(cat requirements.txt); do pip install $req; done
 
cd ..

vi Makefile

make all
make test
make runtest
make pycaffe  
make distribute
 
The binary models can be download with the following script. In caffe-master directory, 

cd scripts
./download_model_binary.py ../models/bvlc_alexnet/
./download_model_binary.py ../models/bvlc_googlenet/
./download_model_binary.py ../models/bvlc_reference_caffenet/
./download_model_binary.py ../models/bvlc_reference_rcnn_ilsvrc13/
./download_model_binary.py ../models/finetune_flickr_style/
 
 
 
Python
sudo apt-get install python-tk 



 
 

No comments:

Post a Comment