ubuntu16.04利用docker搭建cuda+cudnn环境

1.update docker to 19.03
2,docker pull nvidia/cuda ##cuda=10.2,cudnn=7
3,docker run --gpus all -it nvidia/cuda /bin/bash
4,安装编译caffe环境依赖
apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-s erial-dev protobuf-compiler

apt-get install libboost-all-dev #安装boost库

apt-get install libopenblas-dev liblapack-dev libatlas-base-dev

apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

apt-get install git build-essential cmake-gui #注意这里没有cmake

5,安装caffe依赖，但要注意，对于cmake的安装，没有直接使用apt-get来安装，因为用它安装最新版本只能到3.10.0，如果使用它来编译，会由于CUDA版本太新，cmake的时候找不到Cuda的某个库。

cmake要求版本>=3.12.0，安装方式如下：

wget https://cmake.org/files/v3.13/cmake-3.16.5.tar.gz #没有wget的话用apt-get安装(https://cmake.org/download/)
tar -zxvf cmake-3.16.5.tar.gz
cd cmake-3.16.5
ln -s cmakePath/bin/* /usr/bin #直接将cmake中的二进制文件拷贝到/usr/bin下就能用

6,git clone https://github.com/BVLC/caffe.git
cd caffe
#修改CMakeLists.txt
将set(python_version “2” CACHE STRING “Specify which Python version to use”)
改为set(python_version “3” CACHE STRING “Specify which Python version to use”)

cd cmake
#修改cuda.cmake文件
将gpu_archs改为如下：
Known NVIDIA GPU achitectures Caffe can be compiled for.
#This list will be used for CUDA_ARCH_NAME = All option
set(Caffe_known_gpu_archs “30 35 50 60 61”)
#否则或make的时候会出现nvcc fatal : Unsupported gpu architecture 'compute_20’错误

7.#在caffe目录下
mkdir build
cmake …
make -j’nproc’
make install

8,docker保存修改后的镜像
example centos:
start container:[root@afcaf46e8305 /]#docker run -it centos /bin/bash
make some changes:[root@afcaf46e8305 /]#yum install -y vim
exit container: [root@afcaf46e8305 /]#exit

注意afcaf46e8305是产生的容器ID
save the changes:docker commit afcaf46e8305 centos-vim
查看镜像centos-vim: docker images | grep centos-vim
查看镜像的详细信息：docker inspect centos-vim:afcaf46e8305
使用centos-vim这个镜像:docker run -it centos-vim /bin/bash
查看vim:vim --version