史上最も完全なcaffe環境構築チュートリアル

5131 ワード

きれいな新しいシステムからcaffeハードウェアの構築を開始する:GPU GTX 1060 6 G OS:ubuntu 16.04 x64
システムのインストール後、再起動後、起動画面でfb switching to nouveaufb from efi vgaというエラーが発生した場合は、グラフィックスカードを抜いてグラフィックスドライブをインストールし、グラフィックスカードを挿入すればよい
グラフィックスドライバのインストール:ここでドライババージョン(私が今使っているバージョンは384)を見つけて、このコマンドでドライバsudo apt-get install nvidia-384をインストールします.エラーがあれば、sudo apt-get upgradeを実行してからインストールドライバのコマンドを実行できます.
CUDA:ここでインストールsudo sh ./cuda_8.0.61_375.26_linux.runをダウンロードして、それからしばらく待って、End User License Agreementを読んで直接CTRL+Cを押してインストールしましょう、ヒントに従ってインストールしましょう、少し注意して、再びインストールして駆動する時NOを選ぶことをヒントにして、私達はすでにインストールが成功した後に
sudo vi /etc/ld.so.conf
# 
/usr/local/cuda/lib64/
# 
sudo ldconfig

各種ツールとライブラリのインストール
sudo apt install cmake
sudo apt install unzip
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install  libleveldb-dev
sudo apt-get install libsnappy-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt install protobuf-compiler
sudo apt-get install libprotobuf-dev
sudo apt install python-pip

コンパイルインストールopencv 3
wget https://codeload.github.com/opencv/opencv/zip/3.4.0
unzip opencv-3.4.0.zip
cd unzip opencv-3.4.0.zip
mkdir x64
cd x64
cmake ../  #ippicv_2017u3_lnx_intel64_general_20170822  , ,  
make
sudo make install

GPUがあれば、コンパイルが長いので、何時間もコンパイルします
caffe:python関連のライブラリをインストールすると、iPythonがインストールに失敗する可能性があります.具体的に何をしているのか私にはわかりませんが、無視します.
git clone https://github.com/BVLC/caffe.git
cd caffe/python/
for req in $(cat requirements.txt); do sudo pip install $req; done

caffeのソースディレクトリに進む
cp Makefile.config.example Makefile.config
vi Makefile.config
#  OPENCV_VERSION := 3  # 
#  INCLUDE_DIRS :=   /usr/include/hdf5/serial  
# LIBRARY_DIRS   /usr/lib/x86_64-linux-gnu/hdf5/serial
# 
make pycaffe
make all
make test
make runtest

コンパイル中に現れたら
E0104 10:41:13.060823  3649 common.cpp:114] Cannot create Cublas handle. Cublas won't be available.
E0104 10:41:13.064980  3649 common.cpp:121] Cannot create Curand generator. Curand won't be available.
F0104 10:41:13.065028  3649 syncedmem.hpp:22] Check failed: error == cudaSuccess (30 vs. 0)  unknown error

実行sudo ldconfig /usr/local/cuda/lib64最後に
vi ~/.bashrc
#  export PYTHONPATH=/home/dyjun/deeplearn/caffe/python/
# 
source ~/.bashrc

大功を成し遂げた~