Centos7服务器一般用户安装tensorflow-GPU版本

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

  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

你可能感兴趣的:(python学习,TensorFlow,linux)