您的位置 首页 PyTorch 教程

PyTorch 的 Docker 镜像

PyTorch入门实战教程

Image result for docker

在使用这个镜像前,首先要保证已经安装了 Nvidia 的驱动,并且安装了 nvidia-docker2 来支持 GPU 对 Docker 的读写。

Docker Hub 地址

已经编译好的 Docker Image 在 Docker Hub 上,地址是 https://hub.docker.com/r/anibali/pytorch/。例如,你想使用 CUDA 9.2 版本,则使用下列命令:

现在支持的版本有:

Image tagCUDAPyTorch
no-cudaNone0.4.1
cuda-9.29.20.4.1
cuda-9.09.00.4.1
cuda-8.08.00.4.1

下面的版本也有,但不再提供支持:

Image tagCUDAPyTorch
cuda-9.19.10.4.0
cuda-7.57.50.3.0

如何运行 PyTorch?

具体运行方法请查看 DockerHub 上的详细说明。

命令行示例:

PyTorch入门实战教程

发表评论

电子邮件地址不会被公开。 必填项已用*标注

返回顶部