rocm/pytorch rocm6.4_ubuntu24.04_py3.12_pytorch_release_2.3.0

The PyTorch on ROCm Docker images

huggingface/transformers-pytorch-deepspeed-latest-gpu latest

projectmonai/monailabel 0.8.0

huggingface/transformers-pytorch-deepspeed-latest-gpu-push-ci latest

navarrepratt/pytorch_mpi_mnist 13

huggingface/transformers-pytorch-deepspeed-nightly-gpu latest

navarrepratt/stable-diffusion-trt 8.5.2

yuewuust/nvidia-pytorch 1.12.0-py3-our-v4

mithilaidocker/audiotranscribe master

navarrepratt/pytorch_dist_resnet50 14

yaoyuh/ngc_x86_daa 22.12_03_fiftyone

Images for x86 based on NGC images. For DAA project.

navarrepratt/pytorch_mpi_resnet50 cuda-12.0.1

huggingface/transformers-pytorch-deepspeed-latest-gpu-test latest

zuojianghua/stable-diffusion-docker-image 20230505

esw123/ubuntu20.04-py38-rocm5.4 2023-02-20

t9kpublic/nanogpt test-sdk-v2

nkeetha/daa 22.12

t9kpublic/nvidia-pytorch 23.02-py3

xilinx/vitis-ai-pytorch-rocm ubuntu2004-3.5.0.306

smartpetdocker/jetpack5 gait_analysis

chawins/prep python3.8

azwellplus/deepspeed 1.2

wcarroll/pytorch-sleep latest

yyangatdocker/pytorch-k8s-slurm latest

oneflowinc/onediff torch2.3-cuda11.8

lizelive/sd latest

rainbowsecret/pytorch-13 20230420

soar97/sherpa_server 22.12

patoalejor/pytorch dev-scms

lizelive/dream-fusion latest

esw123/ubuntu20.04-py38-cuda11.8-tensorrt8.5-cudnn8.8-nccl2.16 2023-02-19

lizelive/stable-diffusion-webui latest

rlekeyii/strong_yolov7 latest

peakji92/chatglm 6b

shivanshjain/nvidia-pytorch ftest-22

boon1987/pytorch_lung_disease v1

navarrepratt/tensorrt 20220117.1

aniruok/zs-ucdir v3

jyun24/dktorch-vision latest

sermengi/nvidia-jupyterlab v1

navarrepratt/pytorch_dist_imagenet 1

sunyanan/pytorch-mlbench 23.01-py3

theairlab/tartan-slam focal-noetic-pytorch-1.13.0

Docker environment for running Tartan-SLAM

trentaml/belle-7b-2m 202303252015

linnanwang/lora v6

nostalld/gfootball triton

lae01/customdockerimage 12.13

brookluo/img-spec-ml nvcr-22.09-py3

neilmehta87/cosmoflow_pyt 23.07

surajpaib/monai_bundle_inference latest

tuapuikia/tts latest

reimage/condition latest

itdevwu/pytorch ngc-23.04-py3

Pytorch image derived from the official image, with some additional tools installed and configured.

zz217/deepcam-cpu latest

sfarrell/opencatalyst-opt 23.07.01

tuapuikia/visual-chatgpt 1

aciborowska/mh_eval latest

muhandre/pytorch 23.04-py3-yolov5-gpu

eason2023/mlperf-nvidia language_model_pytorch_22.09code

determinedai/environments-dev cuda-11.3-pytorch-1.10-gpt-neox-deepspeed-gpu-0.30.1

[DO NOT USE] determined-ai environments - development

rocm/pytorch-nightly 2024-09-21-rocm6.2

This docker repo is used for hosting pytorch-nightly-build docker images.

hyperonym/basaran 0.21.1

Open-source alternative to the OpenAI text completion API.

k7212519/stable-diffusion-webui rx7000

stable-diffusion-webui onekey deploy GUI for AMD GPU https://github.com/k7212519/ksd-launcher

chenhsuanlin/neuralangelo 23.04-py3

frosting/onf-server-sd 2.3.0

hxssgaa/torchenv latest

simonebendazzoli/nndet 1.2.3-interactive

rocm/oai-triton preview_2024-06-11_285

The OpenAI Triton docker container with ROCm support.

nasanccs/pytorch-caney dev

amole2e/nvidia-pytorch 23.06-py-03

yibolu96/lyra-chatglm-env 0.0.1

ellisdg/unet3d latest

andreimihai/vllm-rest 0.2.6

chopin/llama2-flyte 0.0.4

songhun/summarization 0.0.2

episodeyang/isaacgym 2023-09-03

nealpu/torch1.13 latest

dongliniris/pytorch 23.07-py3-donglin

raptai/pytorch_horovod rapt

awiss/nemo v20230516

agmalanco/nemomegatron 2gpus-1node-20ts-4microbatch

warrents/trt_benchmark ngc_torch-22.12-py3-torchtrt

mo12896/visual-perma-tracker 0.0.33

t9kpublic/deepspeed-neox 23.02-0.10.0-230914

c00cjz00/nvflare_monai-toolkit 1.0.1-1

ruoqian/tensorrt_llm dev-23.10-py3-trt9.2.0.5

shadhil24/sd-api-container v1.0

dennismalmgren/berzdev latest

t9kpublic/megatron-deepspeed 23.06-py3

strekalov/yoloinference 228322