1.创建容器
docker run --name=容器名 -v 原来目录:映射后的目录 --gpus all -itd nvcr.io/nvidia/pytorch:版本-py3
比如:
docker run --name=xyz -v /home/niu/xyz:/home/xyz --gpus all -itd nvcr.io/nvidia/pytorch:20.12-py3
容器名称就是xyz,原来的目录/home/niu/xyz在容器中映射为/home/xyz,容器版本为20.12。
容器的版本对应着配套的pytorch、cuda、tensorRT、还有ubuntu型号。
可以进入下列网址挑选容器版本:
在创建容器时,注意下述命令中的--rm,这个参数加上之后,停止就会将容器删除,警惕使用:
docker run --name=容器名 -v 原来目录:映射后的目录 --gpus all -itd --rm nvcr.io/nvidia/pytorch:版本-py3
2.查看容器
docker ps # 查询正在运行的容器
docker ps -a # 查询全部容器
3.进入容器
sudo docker exec -it 容器名 bash
4.启动、停止、重启容器
docker start/stop/restart 容器名/容器ID
5.删除容器
docker rm 容器名/容器ID #停止才能删除或者-f强制删除