PyTorch Docker#
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Docker#
docker run --gpus all --rm -ti --ipc=host pytorch/pytorch
docker run -d --name pytorch-notebook -p 8888:8888 -v $PWD/pytorch-notebook:/home/jovyan/work quay.io/jupyter/pytorch-notebook
Screenshots#
