/pytorch-jupyter-docker

Docker image for Jupyter Notebook with PyTorch

Primary LanguageDockerfileMIT LicenseMIT

pytorch-jupyter-docker

Docker image for Jupyter Notebook, SSH with PyTorch, CUDA, etc.

Get Started

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