Docker配置Pytorch深度学习环境

拉取镜像

$ docker pull pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel

查看本地已有镜像

$ docker images

创建容器

$ docker run -it -v /home/cenjw/dataset/:/home/data \
> -v /home/cenjw/kt/:/home/ktmodel --gpus=all --name=torchenv \
> pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel /bin/bash

• -v /home/cenjw/dataset/:/home/data 挂载宿主机数据目录:容器目录

查看已经创建的容器

$ docker ps -a

启动并进入容器

$ docker start id

$ docker attach 容器id

docker常见命令

docker run -it 容器名 /bin/bash 进入容器
exit           退出容器
CTRL + P + Q   退出容器不停止    

删除容器
	docker rm 容器id
	docker rm -f $(docker ps -aq)   // all
	docker ps -a -q | xargs docker rm   // all

docker start id    启动容器
docker stop id     停止容器
docker kill id     强制停止		    

进入当前正在 运行的容器
	docker exec -it 容器id bashShell
	docker attach id

docker logs -tf --tail 10 id   显示日志
docker inspect id              查看元数据