puzlcloud/caffe2 repository overview

⁠caffe2 Docker image

Caffe2 ML framework with various python runtime.

Non-root Docker image used by Puzl Kubernetes cloud⁠ service. Based on official cuda⁠ Docker image.

⁠Supported languages and interpreter versions

• python3.10
• python3.11

⁠Installed packages

⁠OS

• caffe2
• rclone
• conda
• openssh-server
• rsync
• cuda, version 11.8

⁠Python

• torch⁠, version 2.1.0
• torchvision⁠, version 0.16.0
• torchaudio⁠, version 2.1.0
• jupyterlab⁠, version 4.0.7