前置条件:
服务器上已经有nvidia-docker
1、创建容器
stu@XXX:~$ docker images #查看已有镜像
#可以看到Image ID 还有Repository 和tag
#创建容器的容器的时候 可以用ImageID 或者 Repository:Tag 如:"pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel"
stu@XXX:~$ docker run -it --gpus all --name ymz --shm-size="2g" -p 8088:99 -v /data/ymz:/data/ymz pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel /bin/bash
-it表示交互式运行容器;
–name指定容器名;
–shm-size设置的是共享内存大小&