runpod/stack repository overview

This is a convenience image written for the RunPod platform. It is built using the lambda lab open source docker file.

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

It includes:

• Ubuntu 20.04 base
• Nvidia CUDA 11.6
• Nvidia cuDNN
• Pytorch 1.11.0
• Tensorflow 2.8.0
• Optional SSH Daemon running on port 22
• Optional JupyterLab running on 8888
• Bunch of ML utilities

Image configuration via ENV variables:

PUBLIC_KEY: This will set your public key into authorized_keys in ~/.ssh so you don't have to manually add it. It will also launch openssh daemon listening on port 22

JUPYTER_PASSWORD: This allows you to pre-configure the jupyter lab password that is used to auth. It will also start jupyter lab listening on port 8888