tensorflow gpu安装_tensorflow-gpu安装配置

1 介绍

　　目前，主流版本是tensorflow-gpu 1.12.0，线上安装这个版本，和这个版本搭配的cuda是9.0版本，cudnn版本是7.0.5。

2 cuda和cudnn安装

cuda驱动地址：cuda_9.0.176_384.81_linux.run

cudnn库地址: cudnn-9.0-linux-x64-v7.solitairetheme8

2.1 cuda安装

cuda安装sh，安装过程中有几个选项需要输入或者选择，可以直接选择非交互式安装，都使用默认选项。

非交互式安装: sh cuda_9.0.176_384.81_linux.run --silent

交互式安装: sh cuda_9.0.176_384.81_linux.run

2.2 cudnn安装

mv cudnn-9.0-linux-x64-v7.solitairetheme8 cudnn-9.0-linux-x64-v7.tgz

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include

$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

$ sudo chmod a+r /usr/local/cuda/include/cudnn.h

3 环境变量配置

编辑~/.bashrc添加以下环境变量，添加后执行source ~/.bashrc

export PATH=/usr/local/bin:$PATH
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/lib/x86_64-linux-gnu:${LD_LIBRARY_PATH}
export PATH="$CUDA_HOME/bin:$PATH"
export CUDA_VISIBLE_DEVICES=0
export PYTHON_BIN_PATH=/usr/bin/python3

export CUDA_VISIBLE_DEVICES=0

4 安装tensorflow-gpu

sudo pip3 install tensorflow-gpu==1.12.0

安装后测试:

import tensorflow as tf

with tf.Session() as sess:

sess.list_devices()

参考地址

• https://docs.nvidia.com/cuda/archive/9.0/cuda-installation-guide-linux/index.html

总结

以上是生活随笔为你收集整理的tensorflow gpu安装_tensorflow-gpu安装配置的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。