手把手教你安装深度学习软件环境(附代码)

  1. 云栖社区>
  2. 数据派THU>
  3. 博客>
  4. 正文

手把手教你安装深度学习软件环境(附代码)

技术小能手 2017-10-09 14:43:40 浏览3316
展开阅读全文

为了进行强化学习研究,我最近购置了一台基于 Ubuntu 和英伟达 GPU 的深度学习机器。尽管目前在网络中能找到一些环境部署指南,但目前仍然没有全面的安装说明。另外,我也不得不阅读了很多文档来试图理解安装细节——其中的一些并不完整,甚至包含语法错误。因此,本文试图解决这个问题,提供一个详尽的软件环境安装指南。


本文将指导你安装


  • 操作系统(Ubuntu)

  • 4 种驱动和库(GPU 驱动、CUDA、cuDNN 和 pip)

  • 5 种 Python 深度学习库(TensorFlow、Theano、CNTK、Keras 和 PyTorch)


这些软件之间的互相依赖关系如下图所示。其中,你只需要安装一种深度学习库,请自由选择。


9bae25177216803ba6c3aa361c74f88c39fafd1d

深度学习软件安装和依赖示意图


以下是各个软件的细节介绍:


  1. Ubuntu(v16.04.3)——操作系统,各种任务处理。

  2. Nvidia

网友评论

登录后评论
0/500
评论
技术小能手
+ 关注
所属云栖号: 数据派THU