centos7 cuda测试_使用docker（非Nvidia-docker）在基于CentOS image的容器中测试cuda

1，找到宿主机的nvidia驱动设备

gemfield@ai:~$ ls -la /dev | grep nvidia

crw-rw-rw- 1 root root 195, 0 10月 27 13:31 nvidia0

crw-rw-rw- 1 root root 195, 1 10月 27 13:31 nvidia1

crw-rw-rw- 1 root root 195, 255 10月 27 13:31 nvidiactl

crw-rw-rw- 1 root root 195, 254 10月 27 13:31 nvidia-modeset

crw-rw-rw- 1 root root 240, 0 10月 27 13:31 nvidia-uvm

crw-rw-rw- 1 root root 240, 1 10月 27 13:31 nvidia-uvm-tools

2，起docker并挂载nvidia驱动设备

gemfield@ai:~$ docker run -itd --name centos -v nvidia_driver_384.90:/usr/local/nvidia:ro --device /dev/nvidia0:/dev/nvidia0 --device /dev/nvidiactl:/dev/nvidiactl --device /dev/nvidia-uvm:/dev/nvidia-uvm --device /dev/nvidia1:/dev/nvidia1 --device /dev/nvidia-uvm-tools:/dev/nvidia-uvm-tools gemfield/centos:7.3 bash

3，在docker容器中安装gcc c++编译器

[root@db041e9301af /]# yum install gcc gcc-c++

4，安装bzip2，为了Anaconda

[root@db041e9301af ~]# yum install bzip2.x86_64

5,安装外部源用来安装nvidia需要的商业驱动

[root@db041e9301af ~]# yum install epel-release

[root@db041e9301af ~]# yum install dkms

6, 下载cuda

#cuda 8.0

[root@db041e9301af ~]# wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda-repo-rhel7-8-0-local-8.0.44-1.x86_64-rpm

7，安装cuda

[root@db041e9301af ~]# rpm -i cuda-repo-rhel7-8-0-local-8.0.44-1.x86_64-rpm

[root@db041e9301af ~]# yum clean all

[root@db041e9301af ~]# yum install cuda

8, 安装ffmpeg

rpm --import http://li.nux.ro/download/nux/RPM-GPG-KEY-nux.ro

rpm -Uvh http://li.nux.ro/download/nux/dextop/el7/x86_64/nux-dextop-release-0-5.el7.nux.noarch.rpm

yum install ffmpeg ffmpeg-devel -y

9，安装gtk

yum install gtk+-devel gtk2-devel

10，编译opencv

yum install cmake

rm -f /usr/bin/python

ln -s /opt/tool/anaconda3/bin/python /usr/bin/python

cmake -DWITH_CUDA=off -DCMAKE_INSTALL_PREFIX=/opt/tool/gemfield/opencv/ -DPYTHON_INCLUDE_DIR=$(python -c "from distutils.sysconfig import get_python_inc; print(get_python_inc())") -DPYTHON_LIBRARY=$(python -c "import distutils.sysconfig as sysconfig; print(sysconfig.get_config_var('LIBDIR'))") ../ 2>&1 | tee /tmp/gemfield1

11, 编译Caffe

yum install protobuf-devel hdf5-devel leveldb-devel opencv-devel

yum install boost-devel gflags-devel glog-devel lmdb-devel

yum install atlas-devel snappy-devel

#库的版本名字似乎有变化，建立以下软链接

ln -s /usr/lib64/atlas/libtatlas.so /usr/lib64/atlas/libatlas.so

ln -s /usr/lib64/atlas/libtatlas.so /usr/lib64/atlas/libcblas.so

#python库，这个是import caffe的时候需要的

pip install scikit-image numpy protobuf

#这个faster-rcnn需要

pip install easydict

pip install pyyaml

yum install tkinter

(不定期更新)