Ubuntu 16.04 安装深度学习环境 GPU 加速版
来源:互联网 发布:整理相册的软件 编辑:程序博客网 时间:2024/04/29 11:31
深度学习环境搭建起来比较麻烦,特别是 GPU 加速版。既要安装 Nvidia 的驱动,又要安装 cuda 和 cuDNN 。本文通过 docker 容器技术搭建基于keras (tensorflow) 的深度学习环境。
Docker 是一种容器技术,将你要运行的任何环境只通过几行命令就可以搭建完成。而 Docker 中除了用普通的 docker pull 方式直接拉去现成的深度学习镜像来搭建深学环境外,还可以通过用 Dockerfile 来完成这一任务。
经过笔者的不(jian)懈(chi)努(cai)力(keng),一个完整的经过测试的深度学习 Dockerfile 见以下链接:
https://github.com/zhudaoruyi/deep-learning-gpu-env
从 GitHub 上 clone 完后,(记住给个 Star )进入到 Dockerfile 所在的目录。
Dockerfile 构建镜像的方法
docker build -t name:tag .
运行该镜像的方法gpu加速)运行该镜像的方法(GPU加速)
nvidia-docker run -d -p 8888:8888 --name test -v /home/pzw:/home/workspace 镜像IDnvidia-docker exec -it 容器ID
保存该镜像的方法)保存该镜像的方法
docker save -o 镜像名称
保存该容器的方法
docker export -o 容器名称
nvidia-docker 的安装
参考 https://github.com/zhudaoruyi/nvidia-docker
注意:安装 nvidia-docker 之前先安装好 docker
为了确认 nvidia-docker 是否安装成功,运行
nvidia-docker run --rm nvidia/cuda nvidia-smi
如果正确输出了本机的 GPU 信息,则安装成功。
例如:
+-----------------------------------------------------------------------------+| NVIDIA-SMI 367.48 Driver Version: 367.48 ||-------------------------------+----------------------+----------------------+| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC || Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. ||===============================+======================+======================|| 0 Tesla M40 24GB Off | 0000:02:00.0 Off | 0 || N/A 33C P0 57W / 250W | 22427MiB / 22939MiB | 0% Default |+-------------------------------+----------------------+----------------------+| 1 Tesla M40 24GB Off | 0000:82:00.0 Off | 0 || N/A 37C P0 58W / 250W | 21663MiB / 22939MiB | 0% Default |+-------------------------------+----------------------+---------------------
阅读全文
1 0
- Ubuntu 16.04 安装深度学习环境 GPU 加速版
- 用GPU加速深度学习: Windows安装CUDA+TensorFlow教程
- 使用GPU和Theano加速深度学习
- 使用GPU和Theano加速深度学习
- 深度学习框架Theano利用GPU加速
- 【深度学习】win10+Theano+GPU加速配置
- 深度学习_caffe-ubuntu-GPU 配置环境(0)
- supervessel-免费云镜像︱GPU加速的Caffe深度学习开发环境
- 集群服务器环境下安装Caffe深度学习库(GPU)
- TensorFlow深度学习入门——Win10下安装Tensorflow和GPU加速
- 深度学习篇——Ubuntu16.04下安装caffe(GPU加速)
- 使用Amazon AWS搭建GPU版tensorflow深度学习环境
- 深度学习平台Caffe环境搭建【GPU版】
- 深度学习平台TensorFlow环境搭建【GPU版】
- 深度学习平台Theano环境搭建【GPU版】
- 深度学习之五:使用GPU加速神经网络的训练
- mxnet:结合R与GPU加速深度学习
- 深度学习卷积算法的GPU加速实现方法
- Oracle数据库相关概念
- 单例模式&多例模式
- LeetCode 376. Wiggle Subsequence
- 记一次 gitlab 与老的 nginx 冲突处理
- 远程访问linux图形化界面
- Ubuntu 16.04 安装深度学习环境 GPU 加速版
- 当excel表格单元格的格式是日期格式非文本格式的时候,phpexcel 应该要这样处理(thinkphp3.2)
- 【Gerrit】Linux+Gerrit+Nginx配置全过程
- 文章标题
- Nginx配置反向代理访问 Gitlab
- MYSQL性能优化之Mysql数据库高可用架构设计之MHA架构设计(下)
- 给本地openwrt/LEDE软件源添加软件包,更新Package.sig签名文件,解决Signature check failed问题
- cvGetSize与cvSize的区别
- Deep Learning(深度学习)学习笔记整理系列