PyTorch Docker

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Docker

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

Screenshots

https://pytorch.org/tutorials/_static/img/thumbnails/tensorboard_scalars.png

https://pytorch.org/tutorials/_images/mp_vs_rn_vs_pp.png

https://pytorch.org/tutorials/_images/split_size_tradeoff.png