1. 參考https://nvidia.github.io/nvidia-container-runtime/ 配置nvidia 源
主要執行:
curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \ sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \ sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update
2. 參考https://gitee.com/wuxler/nvidia-container-runtime?_from=gitee_search 安裝nvidia-container-runtime
主要執行:
安裝運行時
sudo apt-get install nvidia-container-runtime
停止docker
systemctl stop docker
把運行時添加到docker中：
dockerd --add-runtime=nvidia=/usr/bin/nvidia-container-runtime
創建一個容器:
創建容器時一定要指定gpus參數
docker run -it --gpus=all 鏡像名 執行命令
進入容器后執行
nvidia-smi
如果正常會有GPU信息
PS： 創建容器時偌提示:
docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]]. ERRO[0000] error waiting for container: context canceled
apt-get install nvidia-container-toolkit
systemctl restart docker