Sign inSign up

runpod/pytorch

By runpod

Updated 1 minute ago

Image
21

10M+

runpod/pytorch repository overview

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

Tag summary

Content type

Image

Digest

sha256:16cc285c5

Size

11.2 GB

Last updated

1 minute ago

docker pull runpod/pytorch:1.4.0-rc.164-cu1281-torch2130-ubuntu2204