Ceci est un mémo de travail jusqu'à ce que vous puissiez exécuter Keras (version GPU) et OpenCV 3.2 sur Python 2.7.
https://developer.nvidia.com/rdp/cudnn-download
À partir de «Télécharger cuDNN v6.0 (27 avril 2017), pour CUDA 8.0» Vous devez télécharger "cuDNN v6.0 Library for Linux". (cuDNN v7.0 ne semble pas être pris en charge par TensorFlow 1.3.0 [^ 1])
Supposons que vous placiez le fichier téléchargé dans "/tmp/cudnn-8.0-linux-x64-v6.0.tgz".
Comment installer nvidia-docker est décrit dans le lien ci-dessous.
https://github.com/NVIDIA/nvidia-docker
nvidia-docker pull nvidia/cuda
nvidia-docker run -it -v /tmp:/tmp nvidia/cuda bash
apt-get update -y
apt-get upgrade -y
apt-get install -y python-dev python-setuptools curl
curl -kL https://bootstrap.pypa.io/get-pip.py | python
pip install tensorflow-gpu==1.3.0
tar -xvzf /tmp/cudnn-8.0-linux-x64-v6.0.tgz
mv cuda/lib64/libcudnn* /usr/local/cuda/lib64/
mv cuda/include/cudnn.h /usr/local/cuda/include/
chmod a+r /usr/local/cuda/include/cudnn.h
chmod a+r /usr/local/cuda/lib64/libcudnn*
echo "export LD_LIBRARY_PATH='$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64'" >> ~/.bashrc
echo "export CUDA_HOME=/usr/local/cuda" >> ~/.bashrc
source ~/.bashrc
pip install Keras==2.0.6
python -c "import keras; print(keras.__version__)"
Si «2.0.6» est affiché, il réussit.
Tout le reste se fait à l'intérieur du conteneur Docker. Je me réfère à l'article [^ 2].
apt-get install -y build-essential cmake git pkg-config libjpeg8-dev libtiff5-dev libjasper-dev libpng12-dev libavcodec-dev libavformat-dev libswscale-dev libv4l-dev libxvidcore-dev libx264-dev libgtk-3-dev libatlas-base-dev gfortran python2.7-dev zip libhdf5-dev libprotobuf-dev libprotoc-dev protobuf-compiler xserver-xorg doxygen wget
Au milieu
Country of origin for the keyboard: 50
Keyboard layout: 1
Un message apparaîtra, vous devez donc le sélectionner en fonction de votre propre environnement. Dans mon cas, je l'ai réglé sur 50 (japonais) et 1 (japonais).
wget -O opencv.zip https://github.com/Itseez/opencv/archive/3.2.0.zip
unzip opencv.zip
cd opencv-3.2.0
mkdir build
cd build
cmake -D CMAKE_BUILD_TYPE=RELEASE -D BUILD_CUDA_STUBS=OFF -D CMAKE_INSTALL_PREFIX=/usr/local -D WITH_FFMPEG=OFF -D BUILD_opencv_python2=ON -D BUILD_EXAMPLES=OFF -DBUILD_opencv_dnn=OFF ..
make -j $(nproc)
make install
Le code [^ 3] peut être utilisé comme référence pour les options de cmake.
python -c "import cv2; print(cv2.__version__)"
Si «3.2.0» est affiché dans, cela réussit.
References
Recommended Posts