docker.io/hpcaitech/pytorch-cuda
PyTorch CUDA 容器镜像
这是一个基于 Docker 的容器镜像,包含了 PyTorch 和 CUDA 运行时环境。您可以使用该镜像来运行 PyTorch 代码,并充分利用 GPU 加速计算能力。
该镜像包含以下内容:
PyTorch
CUDA 运行时
基础依赖库
您可以使用以下命令拉取该镜像:
docker pull docker.io/hpcaitech/pytorch-cuda
您可以通过以下方式访问该镜像:
https://hub.docker.com/r/hpcaitech/pytorch-cuda
下载:1
版本:1