Docker for Pytorch

1. Docker

1.1 安装

官网教程

1.2 添加用户

If you want to run docker as non-root user then you need to add it to the docker group.

Create the docker group if it does not exist

$ sudo groupadd docker

Add your user to the docker group.

$ sudo usermod -aG docker $USER

2. nvidia-docker for GPU

2.1安装

官网教程

2.2报错 no swap limit support

sudo vim /etc/default/grub
	GRUB_CMDLINE_LINUX="cgroup_enable=memory swapaccount=1"
sudo update-grub
sudo reboot

3.Pytorch

docker run --gpus all --rm -ti --ipc=host pytorch/pytorch:latest

参考

1.Pytorch镜像安装
2.docker实战