Ubuntu16.04版本搭建Tensorflow环境并启用GPU支持
在配置Tensorflow并开启GPU支持的过程中遇到了一些问题,写这篇文章记录一下
环境搭建的基本流程
- 在ubuntu上安装支持GPU版本的Tensorflow,特别需要注意的是针对Ubuntu16.04系统,Tensorflow中文站上的版本已经有些过时,在开启GPU支持时(安装Cuda合cuDNN)会出现问题,所以最好直接按照Github上面的instruction来安装,地址: Tensorflow-Github
简单翻译一下安装步骤:
- TF需要python2.7或者python3.3+的支持,并且GPU必须时Cuda-capable的,可以在这里查看自己的n卡有没有被Cuda支持GPU
使用pip命令进行安装
$ pip install tensorflow-gpu
如果上面这种安装方式速度太慢,可以采用如下的方式:
针对python2.7: