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

References

• PyTorch Docker

• PyTorch GitHub

• PyTorch

• SINGLE-MACHINE MODEL PARALLEL BEST PRACTICES