NVIDIA Pytorch Docker

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()