0°
PyTorch入门实战教程

PyTorch 的 Docker 镜像

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入门实战教程
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论