pytorchlab/pytorch

•Updated 4 months ago

Minimal PyTorch base images for GPU (CUDA 12.6) & CPU — with optional pre-bundled YOLO weights

Image

Languages & frameworks

Machine learning & AI

Data science

1.7K

Tags for pytorchlab/pytorch

Sort by

TAG

cpu-python3.11-yolo26m-seg

Last pushed 4 months by pytorchlab

docker pull pytorchlab/pytorch:cpu-python3.11-yolo26m-seg

Digest	OS/ARCH	Compressed size
189308b6a98b	linux/amd64	490.22 MB

TAG

cpu-python3.11-yolo26m

Last pushed 4 months by pytorchlab

docker pull pytorchlab/pytorch:cpu-python3.11-yolo26m

Digest	OS/ARCH	Compressed size
6ba055503cea	linux/amd64	475.16 MB

TAG

2.4.1-cpu-py3.11-slim

Last pushed 4 months by pytorchlab

docker pull pytorchlab/pytorch:2.4.1-cpu-py3.11-slim

Digest	OS/ARCH	Compressed size
fc97e351c659	linux/amd64	431.44 MB

TAG

2.9.1-cuda12.6-cudnn9-yolo26m-seg

Last pushed 4 months by pytorchlab

docker pull pytorchlab/pytorch:2.9.1-cuda12.6-cudnn9-yolo26m-seg

Digest	OS/ARCH	Compressed size
ef367c1c0c3a	linux/amd64	3.84 GB

TAG

2.9.1-cuda12.6-cudnn9-yolo26m

Last pushed 4 months by pytorchlab

docker pull pytorchlab/pytorch:2.9.1-cuda12.6-cudnn9-yolo26m

Digest	OS/ARCH	Compressed size
11b99265d39e	linux/amd64	3.83 GB

TAG

2.9.1-cuda12.6-cudnn9-runtime

Last pushed 4 months by pytorchlab

docker pull pytorchlab/pytorch:2.9.1-cuda12.6-cudnn9-runtime

Digest	OS/ARCH	Compressed size
c515970f451e	linux/amd64	3.78 GB

TAG

cpu-python3.11-yolo11m

Last pushed 7 months by pytorchlab

docker pull pytorchlab/pytorch:cpu-python3.11-yolo11m

Digest	OS/ARCH	Compressed size
e4d8e555afaa	linux/amd64	380.39 MB

TAG

2.9.1-cuda12.6-cudnn9-yolo11m

Last pushed 7 months by pytorchlab

docker pull pytorchlab/pytorch:2.9.1-cuda12.6-cudnn9-yolo11m

Digest	OS/ARCH	Compressed size
5193ac90c40f	linux/amd64	3.75 GB