docker.io/diffusers/diffusers-pytorch-cuda

这是一个基于PyTorch和CUDA的Diffusers模型的Docker镜像。它包含了运行Diffusers库所需的一切,包括PyTorch、CUDA驱动程序以及预训练的Diffusers模型。 这意味着你可以方便地在支持CUDA的GPU上运行各种扩散模型,例如Stable Diffusion等,而无需自行安装和配置复杂的依赖项。 该镜像简化了扩散模型的部署和使用,适合用于图像生成、图像编辑等AI应用场景。

下载:9 版本:1

docker.io/diffusers/diffusers-pytorch-cpu

这是一个基于 PyTorch 的 CPU 版本的 Diffusers 模型镜像。它包含了用于图像生成的各种扩散模型,可以在没有 GPU 的情况下运行。 适合在 CPU 上进行图像生成相关的实验和开发。

下载:4 版本:1