1 介紹

　　目前，主流版本是tensorflow-gpu 1.12.0，線上安裝這個版本，和這個版本搭配的cuda是9.0版本，cudnn版本是7.0.5。

2 cuda和cudnn安裝

cuda驅(qū)動地址：cuda_9.0.176_384.81_linux.run

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

2.1 cuda安裝

cuda安裝sh，安裝過程中有幾個選項(xiàng)需要輸入或者選擇，可以直接選擇非交互式安裝，都使用默認(rèn)選項(xiàng)。

非交互式安裝: 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 環(huán)境變量配置

編輯~/.bashrc添加以下環(huán)境變量，添加后執(zhí)行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

總結(jié)

以上是生活随笔為你收集整理的tensorflow gpu安装_tensorflow-gpu安装配置的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。