Dies ist eine Arbeitsnotiz, bis Sie Keras (GPU-Version) und OpenCV 3.2 unter Python 2.7 ausführen können.
https://developer.nvidia.com/rdp/cudnn-download
Aus "Herunterladen von cuDNN v6.0 (27. April 2017) für CUDA 8.0" Sie müssen "cuDNN v6.0 Library for Linux" herunterladen. (cuDNN v7.0 scheint von TensorFlow 1.3.0 [^ 1] nicht unterstützt zu werden)
Angenommen, Sie legen die heruntergeladene Datei in "/tmp/cudnn-8.0-linux-x64-v6.0.tgz" ab.
Wie man nvidia-docker installiert, wird unter dem folgenden Link beschrieben.
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__)"
Wenn 2.0.6 angezeigt wird, ist es erfolgreich.
Der Rest erledigt den Docker-Container. Ich beziehe mich auf den Artikel [^ 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
Mitten drin
Country of origin for the keyboard: 50
Keyboard layout: 1
Die Nachricht wird angezeigt, daher müssen Sie sie entsprechend Ihrer eigenen Umgebung auswählen. In meinem Fall habe ich 50 (Japanisch) und 1 (Japanisch) eingestellt.
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
Der Code [^ 3] kann als Referenz für die Optionen für cmake verwendet werden.
python -c "import cv2; print(cv2.__version__)"
Wenn "3.2.0" in angezeigt wird, ist dies erfolgreich.
References
Recommended Posts