Docker 使用顯卡

注意： Docker 可以共享使用顯卡，即多個docker 可以共享使用同一塊顯卡

環境

• 硬件：Tesla T4 八張顯卡
• 系統：Centos7.9 操作系統
• 服務器安裝完操作系統，並安裝完顯卡驅動
  請參照：https://www.cnblogs.com/lixinliang/p/14705315.html

docker 使用gpu 顯卡方式

• 宿主機查看顯卡，預計使用0號顯卡

shell>  nvidia-smi

• 創建啟動docker 容器的腳本，要求為 32G內存，16核cpu，0號顯卡

#   cat leichuangxin.sh 
PORT=101
USER=leichuangxin
PASS=`uuidgen`

if [[ ! -e /export/$USER ]]; then
    mkdir /export/$USER/.ssh -p
else
    echo "/export/$USER exist"
fi

docker run -itd --name $USER \
           --restart always \
           -v /export/$USER:/export \
           -v /export/$USER/.ssh/:/root/.ssh/ \
           --runtime=nvidia \
           -e NVIDIA_VISIBLE_DEVICES=0 \ #顯卡使用0號
           -m 32G \       #內存分配 32G
           --cpus 16 \    #cpu 核心數分配16c
           -p ${PORT}22:22 \
           -p ${PORT}23-${PORT}25:${PORT}23-${PORT}25 \
           yz.harbor.moviebook.com/base/cuda:11.1.1-runtime-ubuntu18.04
#docker exec -it $USER bash -c "echo root:$PASS|chpasswd"
#docker exec -it $USER bash -c "sed -i 's/PasswordAuthentication yes/PasswordAuthentication no/' /etc/ssh/sshd_config; /etc/init.d/ssh restart"

• 啟動容器

bash leichuangxin.sh 

• 查看容器顯卡是否正常

shell> docker exec -it leichuangxin bash

證明已成功使用宿主機顯卡