Tensors and Dynamic neural networks in Python with strong GPU acceleration
docker run --gpus all --rm -ti --ipc=host pytorch/pytorch
PyTorch Docker
PyTorch GitHub
PyTorch
SINGLE-MACHINE MODEL PARALLEL BEST PRACTICES