torch各个版本镜像_Deepo:几乎包含所有主流深度学习框架的Docker镜像

Deepo是一个包含多种深度学习框架的Docker镜像,提供了一个完整的深度学习研究环境,支持如TensorFlow、PyTorch、Keras等。通过简单步骤安装Docker和nvidia-docker,可以从Docker Hub获取或构建Deepo镜像。使用nvidia-docker运行容器并利用GPU,可在交互式shell中探索各种框架。
摘要由CSDN通过智能技术生成

原文:Deepo

作者:Ming

译者:Teixeira10

【译者注】在本文中,作者简单介绍了Deepo的使用,给出了清晰的步骤和命令,也将各个深度学习框架进行了比较。

以下为译文:

Deepo是一个几乎包含所有流行深度学习框架的Docker映像,拥有一个完整的可复制的深度学习研究环境。它涵盖了当前最流行的深度学习框架:

theano,tensorflow,sonnet,pytorch,keras,lasagne,mxnet,cntk,chainer,caffe,torch。

快速启动

安装

步骤1:安装Docker和nvidia-docker。

步骤2:获得Deepo镜像。

你可以直接从Docker Hub下载镜像,或者自己构建镜像。

选项1:从Docker Hub获得镜像(建议)

docker pull ufoym/deepo1

2

选项2:在本地构建Docker镜像

git clone https://github.com/ufoym/deepo.git

cd deepo && docker build -t ufoym/deepo .1

2

3

请注意,这可能需要几个小时,因为它从头开始编译了一些库。

用法

现在你可以试试这个命令:

nvidia-docker run --rm ufoym/deepo nvidia-smi1

2

这是能够运行的,并且使Deepo能够在docker容器内使用GPU。如果这个方法不起作用,可以搜索nvidia

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值