PyTorch 的 Docker 镜像

清疚 2022-02-13 02:55 737阅读 0赞

## Docker Hub 地址 ##

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

<table> 
 <tbody> 
  <tr> 
   <td> <p>1</p> </td> 
   <td> <p>$ docker pull anibali/pytorch:cuda-9.2</p> </td> 
  </tr> 
 </tbody> 
</table>

现在支持的版本有：

<table> 
 <thead> 
  <tr> 
   <th>Image tag</th> 
   <th>CUDA</th> 
   <th>PyTorch</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td><code>no-cuda</code></td> 
   <td>None</td> 
   <td>0.4.1</td> 
  </tr> 
  <tr> 
   <td><code>cuda-9.2</code></td> 
   <td>9.2</td> 
   <td>0.4.1</td> 
  </tr> 
  <tr> 
   <td><code>cuda-9.0</code></td> 
   <td>9.0</td> 
   <td>0.4.1</td> 
  </tr> 
  <tr> 
   <td><code>cuda-8.0</code></td> 
   <td>8.0</td> 
   <td>0.4.1</td> 
  </tr> 
 </tbody> 
</table>

下面的版本也有，但不再提供支持：

<table> 
 <thead> 
  <tr> 
   <th>Image tag</th> 
   <th>CUDA</th> 
   <th>PyTorch</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td><code>cuda-9.1</code></td> 
   <td>9.1</td> 
   <td>0.4.0</td> 
  </tr> 
  <tr> 
   <td><code>cuda-7.5</code></td> 
   <td>7.5</td> 
   <td>0.3.0</td> 
  </tr> 
 </tbody> 
</table>

## 如何运行 PyTorch？ ##

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

命令行示例：

<table> 
 <tbody> 
  <tr> 
   <td> <p>1</p> <p>2</p> <p>3</p> <p>4</p> <p>5</p> <p>6</p> <p>7</p> </td> 
   <td> <p>docker run --rm -it --init \</p> <p>&nbsp;&nbsp;--runtime=nvidia \</p> <p>&nbsp;&nbsp;--ipc=host \</p> <p>&nbsp;&nbsp;--user="$(id -u):$(id -g)" \</p> <p>&nbsp;&nbsp;--volume=$PWD:/app \</p> <p>&nbsp;&nbsp;-e NVIDIA_VISIBLE_DEVICES=0 \</p> <p>&nbsp;&nbsp;anibali/pytorch python3 main.py</p> </td> 
  </tr> 
 </tbody> 
</table>

[DockerHub]: https://www.pytorchtutorial.com/goto/https://hub.docker.com/r/anibali/pytorch/

发表评论取消回复

表情：

评论列表（有 0 条评论，737人围观）

还没有评论，来说两句吧...

相关阅读

相关为pytorch的docker镜像增加ssh功能

修改dockerfile如下： pytorch-1.4 FROM pytorch/pytorch:1.4-cuda10.1-cudnn7-runtime C

左手的ㄟ右手/ 2022年12月11日 13:32/ 0 赞/ 172 阅读

相关 Docker镜像

镜像是Docker的三大核心概念之一。 Docker运行容器前需要本地存在对应的镜像，如果镜像不存在本地，Docker会尝试先从默认镜像仓库下载，用户也可以通过配置，使用自定

青旅半醒/ 2022年09月30日 13:47/ 0 赞/ 618 阅读

相关 Docker 镜像

一、下载`Kafka` A、官方下载 [http://mirror.bit.edu.cn/apache/kafka/][http_mirror.bit.edu

r囧r小猫/ 2022年09月27日 00:27/ 0 赞/ 193 阅读

相关 docker镜像

docker镜像 docker容器是基于docker 镜像来运行的查看本地镜像 `docker images [options]` options:

ゝ一世哀愁。/ 2022年08月19日 15:15/ 0 赞/ 225 阅读

相关 Docker 镜像

获取镜像获取镜像使用docker pull命令： docker pull NAME\[:TAG\] 如果不显示的指定TAG，则默认会选择latest标签，即下载仓库中最

旧城等待，/ 2022年07月15日 05:25/ 0 赞/ 285 阅读

相关 Docker 镜像

获取镜像获取镜像使用docker pull命令： docker pull NAME\[:TAG\] 如果不显示的指定TAG，则默认会选择latest标签，即下载仓库中最

秒速五厘米/ 2022年07月15日 05:25/ 0 赞/ 292 阅读

相关 Docker 镜像

简介镜像是一种轻量级、可执行的独立软件包，用来打包软件运行环境和基于运行环境开发的软件，它包含运行某个软件所需的所有内容，包括代码、运行时、库、环境变量和配置文件。

秒速五厘米/ 2022年03月17日 15:14/ 0 赞/ 325 阅读

相关 PyTorch 的 Docker 镜像

Docker Hub 地址已经编译好的 Docker Image 在 Docker Hub 上，地址是 https://hub.docker.com/r/anibali

清疚/ 2022年02月13日 02:55/ 0 赞/ 738 阅读

相关 Docker 镜像

一、镜像是什么？ 1、UnionFS（联合文件系统） Union文件系统（UnionFS）是一种分层、轻量级并且高性能的文件系统，它支持对文件系统的修改作为一次提交

谁践踏了优雅/ 2021年09月23日 22:04/ 0 赞/ 434 阅读

相关 Docker 镜像

Docker 镜像是一个特殊的文件系统，除了提供容器运行时所需的程序、库、资源、配置等文件外，还包含了一些为运行时准备的一些配置参数（如匿名卷、环境变量、用户等）。镜像不...

朱雀/ 2020年05月17日 08:32/ 0 赞/ 2719 阅读