第一步
依赖包:yum install -y gcc gcc-c++ kernel-devel kernel-headers
可通过此命令获取离线包:
yum install --downloadonly --downloaddir=/u02/lib gcc gcc-c++ kernel-devel kernel-headers
### 第二步
安装cuda前,先关闭nouveau:
# vim /lib/modprobe.d/dist-blacklist.conf
blacklist nouveau #添加此行
options nouveau modeset=0 #添加此行
# blacklist nvidiafb #将nvidiafb的此行注释掉
第三步
cuda 10.1安装:
./cuda_10.1.168_418.67_linux.run
按提示输入,安装选择全部选择(包括第一个driver)
完成安装后,输入nvidia-smi,有打印显卡信息则驱动安装成功
输入cat /usr/local/cuda/version.txt,有打印cuda版本10.1则cuda安装成功
第三步
cudnn 7.5安装:
解压cudnn-10.1-linux-x64-v7.5.1.10.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
完成后,输入cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2,有打印cudnn版本为7.5则安装成功
第四步
nvidia-docker的安装:docker-ce安装完并启动docker服务后,装nvidia-docker
添加仓库:
~~~
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo
# 安装nvidia-docker2,重载Docker daemon configuration
sudo yum install -y nvidia-docker2
sudo pkill -SIGHUP dockerd
~~~
#### 测试一下是否安装成功
docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi,有打印显卡信息则安装成功