docker.io/nvcr.io/nvidia/cuda

好!下面是这个docker镜像的描述: CUDA (Compute Unified Device Architecture) 是 NVIDIA 开发的一种用于加速 GPU 计算的编程框架。该镜像基于 NVIDIA 的 CUDA 环境,提供了一个完整的 CUDA 工具链,支持开发、调试和部署使用 CUDA 的应用程序。 本镜像包含以下组件: CUDA Toolkit NVIDIA Driver nvidia-smi 其他 NVIDIA 相关工具 该镜像支持多种操作系统,包括 Ubuntu、CentOS 和 Windows。它是开发和部署使用 CUDA 的应用程序的理想选择。

下载:75 版本:1

docker.io/nvcr.io/nvidia/pytorch

这里是镜像的描述信息: NVIDIA PyTorch Docker Image 这是一个基于PyTorch框架的Docker容器镜像,提供了一个完整的深度学习环境。该镜像包含了PyTorch 1.x版本,以及其他必需的依赖包,如CUDA、cuDNN等。 使用这个镜像,您可以轻松地在本地环境中搭建一个深度学习工作站,进行各种机器学习和计算机视觉任务的实验和开发。 此外,该镜像还支持GPU加速,通过NVIDIA的CUDA和cuDNN技术,可以显著提高PyTorch的性能和效率。

下载:61 版本:5