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