简单的Ubuntu16.04 tensorflow, keras环境配置


首先, 请一定用Anaconda装python, 它会很大简化这个过程

如果装CPU版本, 直接

conda install tensorflow

就行了

 

=======================================================================================

GPU版本如下

1. 安装GPU驱动

查看兼容版本

sudo add-apt-repository ppa:graphics-drivers/ppa

执行后, 最上方会出现official release 版本 和 long-lived 版本, 推荐使用后者吧。

sudo apt-get install nvidia-<version>

如: sudo apt-get install nvidia-384

之后安装CUDA

https://developer.nvidia.com/cuda-toolkit-archive

这里可以找到各种版本的CUDA

其实你也可以不用自己下载CUDA, 直接

 

conda install tensorflow-gpu

conda会帮你解决一切的

====================================================================================

然后装cudnn (不存在的!!)

 

直接 conda install tensorflow-gpu 就行了

什么你在逗我?cudnn 都不用?

不好意思, 这

 

就是 Anaconda

 

keras 就 pip install keras就好了


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM