string(21) "{"docs":[],"count":0}" array(2) { ["docs"]=> array(0) { } ["count"]=> int(0) } Centos7服务器一般用户安装tensorflow-GPU版本 - 爱码网

安装tensorflow-GPU首先需要有python,cuda与cudnn,不同的tensorflow版本对应不同的cuda与cudnn,对应如下。
Centos7服务器一般用户安装tensorflow-GPU版本

  1. 首先安装anaconda3(也可以选择python的一般版本,anaconda为python 的集成环境。(安装时,选择自动加入环境变量)
  2. 安装 cuda
  3. 下载cudnn,将cudnn中的include,lib64分别移动到cuda中的include和lib64下。
    在自己用户的 .bashrc 文件中设置环境变量,vim .bashrc
    添加

export LD_LIBRARY_PATH=/home/EXP/cuda9/lib64:$LD_LIBRARY_PATH

  1. 使用清华镜像安装tensorflow-GPU,

pip install tensorflow-gpu==1.9 -i https://pypi.tuna.tsinghua.edu.cn/simple

5.进行测试

python3
import tensorflow as tf

相关文章: