ubuntu 14.04安裝tensorflow-gpu

  • Post author:
  • Post category:其他


记录一下本次安装tensorflow的流程.

因为之前已将安装过caffe框架,所以已经安装CUDA 8.0 ,Nvidia cuDNN等前提环境.只需要在这个基础上安装tensorflow

采用的是anaconda3,tensorflow-gpu==1.2.0,

安装Anaconda3:

Anaconda可以省去很多麻烦,建议采用这种方式;



清华大学开源软件镜像站

下载想的版本;

//安装Anaconda3
bash Anaconda(你的版本).sh
//环境变量
export PATH=(你的安装路径):$PATH //比如 /root/Anaconda3/bin:$PATH
conda list //查看已经安装的库
conda update pip //更新pip

安装anaconda3过程中,默认路径,如果遇到conda命令的问题,只要添加一下:export PATH=/home/jie/anaconda3/bin:$PATH,(注意:要改为自己的路径)


tensorflow安装:

从清华开源软件镜像站下载需要的版本,建议1.2,1.3报错(cuDNN版本太低),网址https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu

不知什么原因,不能在线安装,所以要选择自己需要的版本下载到本地,下载完之后

进入到下载目录:pip install  tensorflow_gpu-1.2.0-cp36-cp36m-linux_x86_64.whl,即可完成安装.

安装过程中出现的问题:

1libcudnn.so.6:cannot open sharedobjectfile: No such file or directory

因为cuDNN版本的问题,选择1.3以下的版本或提高cuDNN版本.



版权声明:本文为weixin_40740785原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。