一、nvidia docker环境和基本操作介绍
(1)docker images查看所有dockers的镜像
(2)docker run -itd -v /home/lxh:/home/lxh --gpus all -e NVIDIA_DRIVER_CAPABILITES=compute,utility,video --name YourContainerName -h YourContainerName bresee/cuda:11.0-trt7.1-devel-1014 bash
(3)关键参数说明
①-v /home/lxh:/home/lxh我们可以将所需的模型、代码、图片数据等放置在个人目录下,再将宿主机上的文件夹挂载入容器中,禁止挂载除个人目录以外的系统目录,禁止直接挂载home目录,docker有root权限,以免造成系统和他人数据的损坏。
②--gpus all 可以挂载自己需要的GPU编号,0、1、2,all可以全部挂载
③-e NVIDIA_DRIVER_CAPABILITES=compute,utility,video设置NV docker功能项,不要修改
④--name YourContainerName 替换为你要启功的容器名称,建议加上个人名称用于区分
⑤-h YourContainerName 替换为你要启功的容器内部的主机名称,和name保持一致即可
⑥bresee/cuda:11.0-trt7.1-devel-1014 使用docker images获取到的REPOSITORY:TAG,指定镜像来源
⑦bash 指定启动shell类型
(4)容器操作指令
service docker start:启动容器服务
docker ps -a :查看此时在后台运行的容器,也可使用nvidia-smi查看gpu的状态<