Docker image for Jupyter Notebook, SSH with PyTorch, CUDA, etc.
docker run --runtime=nvidia \
-e JUPYTER_TOKEN=<PASSWORD> \
[-v <WORKSPACE>:/workspace] \
-p <JUPYTER_PORT>:8888 \
[-p <SSH_PORT>:22] \
[-p <TENSORBOARD_PORT>:6006] -d controlnet/pytorch-jupyter