安裝NVIDIA驅動

在NVIDIA下載對應的驅動安裝文件

1. 安裝依賴環境

   yum install -y pciutils gcc gcc-c++ kernel-devel
   yum update -y kernel
   

2. 屏蔽系統自帶的nouveau

   # 查看命令：
   lsmod | grep nouveau
   
   # 修改dist-blacklist.conf文件：
   vi /lib/modprobe.d/dist-blacklist.conf
   
   # 將nvidiafb注釋掉:
   #blacklist nvidiafb 
   
   # 然后添加以下語句：
   blacklist nouveau
   options nouveau modeset=0
   
   dracut --force
   

3. 重建initramfs image步驟

   mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
   dracut /boot/initramfs-$(uname -r).img $(uname -r)
   

4. 修改運行級別為文本模式

   systemctl set-default multi-user.target
   

5. 重新啟動

   reboot
   
   lsmod | grep nouveau
   # 檢查上面的命令有沒有結果，如果還有結果
   dracut --force
   

6. 運行安裝文件

   chmod +x NVIDIA-Linux-x86_64-410.129-diagnostic.run
   ./NVIDIA-Linux-x86_64-410.129-diagnostic.run
   
   nvidia-smi
   

安裝NVIDIA Docker

1. 安裝Docker

   yum install -y wget
   cd /etc/yum.repos.d/
   wget http://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo
   yum install -y docker-ce
   systemctl start docker
   systemctl enable docker
   
   vi /etc/docker/daemon.json
   {
     "registry-mirrors": ["https://b9pmyelo.mirror.aliyuncs.com"]
   }
   
   systemctl restart docker
   docker verison
   docker info
   

2. 安裝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
   yum clean expire-cache
   yum install -y nvidia-docker2
   systemctl restart docker
   
   # 啟動啟用GPU的CUDA容器；使用--gpus
   docker run --rm --gpus all nvidia/cuda:10.0-base nvidia-smi
   docker run --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all nvidia/cuda:10.0-base nvidia-smi
   
   # 在兩個GPU上啟動啟用GPU的容器
   docker run --rm --gpus 2 nvidia/cuda:10.0-base nvidia-smi
   
   # 在特定GPU上啟動啟用GPU的容器
   docker run --rm --gpus '"device=0"' nvidia/cuda:10.0-base nvidia-smi
   docker run --rm --gpus '"device=0,1"' nvidia/cuda:10.0-base nvidia-smi
   docker run --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=0,1 nvidia/cuda:10.0-base nvidia-smi
   

3. Dockerfiles

   可以通過環境變量在圖像中設置功能和GPU枚舉。如果環境變量是在Dockerfile中設置的，則無需在命令行上進行設置。docker run

   例如，如果要創建自己的自定義CUDA容器，則應使用以下內容：

   ENV NVIDIA_VISIBLE_DEVICES all
   ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
   

   這些環境變量已在NVIDIA提供的CUDA映像中設置。