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 tag CUDA PyTorch
no-cuda None 0.4.1
cuda-9.2 9.2 0.4.1
cuda-9.0 9.0 0.4.1
cuda-8.0 8.0 0.4.1

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

Image tag CUDA PyTorch
cuda-9.1 9.1 0.4.0
cuda-7.5 7.5 0.3.0

如何运行 PyTorch?

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

命令行示例:

PyTorch入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/pytorch-docker/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部