安装tensorflow-GPU首先需要有python,cuda与cudnn,不同的tensorflow版本对应不同的cuda与cudnn,对应如下。
- 首先安装anaconda3(也可以选择python的一般版本,anaconda为python 的集成环境。(安装时,选择自动加入环境变量)
- 安装 cuda
- 下载cudnn,将cudnn中的include,lib64分别移动到cuda中的include和lib64下。
在自己用户的 .bashrc 文件中设置环境变量,vim .bashrc
添加
export LD_LIBRARY_PATH=/home/EXP/cuda9/lib64:$LD_LIBRARY_PATH
- 使用清华镜像安装tensorflow-GPU,
pip install tensorflow-gpu==1.9 -i https://pypi.tuna.tsinghua.edu.cn/simple
5.进行测试
python3
import tensorflow as tf