使用以下命令即可
docker run -itd --name 容器名字 --gpus all -p 服务器端口号:容器端口号 镜像名字
docker run -itd --name imgsea --gpus all -p 6020:80 ubuntu:20.0
下述为在新的服务器上该如何配置docker运行gpu环境
在nvidia网址Repository configuration | nvidia-container-runtime,获取nvidia-container-runtime的安装源
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
安装nvidia-container-runtime
apt-get install nvidia-container-runtime
启动容器时加入gpus选项,all表示gpu全分配
docker run -itd --name 容器名字 --gpus all -p 服务器端口号:容器端口号 镜像名字
docker run -itd --name imgsea --gpus all -p 6020:80 ubuntu:20.04
进入容器后,用以下命令查看显卡
nvidia-smi
出现以下表示显卡可用
1635860136(1).png
网友评论