0. docker: https://docs.docker.com/engine/install/ubuntu/
1. nvidia-docker: https://github.com/NVIDIA/nvidia-docker
2. pytorch cuda docker image: https://hub.docker.com/r/anibali/pytorch
3. 启动pytorch cuda容器:
sudo docker run –gpus all -it anibali/pytorch:cuda-10.1 /bin/bash
4. import torch
torch.cuda.is_available()