This is a convenience image written for the RunPod platform based on the official pytorch image: https://hub.docker.com/r/pytorch/pytorch

GitHub: https://github.com/Run-Pod/containers⁠

Image configuration via ENV variables:

PUBLIC_KEY: This will set your public key into authorized_keys in ~/.ssh. If this is set, SSH daemon will be started on port 22

JUPYTER_PASSWORD: This allows you to pre-configure the jupyter lab password that is used to auth. If this is set, Jupyter lab will be run and hosted on port 8888