Ubuntu16.04深さ学習環境構成——Caffeのインストール(CUDA 8.0+cuDN 6.0)
2091 ワード
CUDA 8について0とcuDNN 6.0、opencv3.1の構成では、前のブログを表示できます. Ubuntu16.04深さ学習環境構成——NVIDIAグラフィックス駆動、CUDA 8.0、cuDNN v6.0 Ubuntu16.04 opencv 3を構成する.1(CUDA 8.0がインストールされている) ブログを参照:https://blog.csdn.net/yhaolpz/article/details/71375762
1、ダウンロード、解凍Caffe
直接行くことを勧めるhttps://github.com/BVLC/caffezip圧縮パッケージをダウンロードします.gitコマンドcloneを使うと遅いからです.
インストールしたいディレクトリにCaffeを解凍します.
2、Makefile.を修正するconfigファイル
解凍したCaffeフォルダに入り、Makefile.config.コピーしてMakefileと名前を変更します.config:
Makefile.を修正するconfigファイル:
次のように変更します.
1)cuDNNの適用:
2)Opencvの適用:
3)Pythonインタフェースを使用する:
4)パスの変更:
3、Makefileファイルの修正
Caffeホームに入り、Makefileファイルを変更し、以下のように変更します.
4、CUDAのhost_を修正するconfigファイル
変更/usr/local/cuda/include/host_config.hファイル:
5、コンパイル
1、ダウンロード、解凍Caffe
直接行くことを勧めるhttps://github.com/BVLC/caffezip圧縮パッケージをダウンロードします.gitコマンドcloneを使うと遅いからです.
インストールしたいディレクトリにCaffeを解凍します.
2、Makefile.を修正するconfigファイル
解凍したCaffeフォルダに入り、Makefile.config.コピーしてMakefileと名前を変更します.config:
sudo cp Makefile.config.example Makefile.config
Makefile.を修正するconfigファイル:
sudo gedit Makefile.config
次のように変更します.
1)cuDNNの適用:
#USE_CUDNN := 1
:
USE_CUDNN := 1
2)Opencvの適用:
#OPENCV_VERSION := 3
:
OPENCV_VERSION := 3
3)Pythonインタフェースを使用する:
#WITH_PYTHON_LAYER := 1
WITH_PYTHON_LAYER := 1
4)パスの変更:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
3、Makefileファイルの修正
Caffeホームに入り、Makefileファイルを変更し、以下のように変更します.
:
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
4、CUDAのhost_を修正するconfigファイル
変更/usr/local/cuda/include/host_config.hファイル:
#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
//#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
5、コンパイル
make all -j8
sudo make runtest -j8