TensorFlow(GPU) 安装

来源：互联网发布：mac上打开flac 编辑：程序博客网时间：2024/06/04 19:54

今天重新装了下系统，用的Ubuntu16.04，准备安装Tensorflow GPU版
安装过程如下：
1. 首先按照官网提示安装Cuda
安装Cuda网站上的步骤安装即可，安装过程中没有出现问题，注意需要设置两个路径PATH和LD_LIBRARY_PATH
之后验证安装是否成功
1) 首先将一些sample文件放在便于修改的路径中

cuda-install-samples-8.0.sh

之后make
我在make时出错了

cannot find -lnvcuvid

在网上查了下，找到了一个解决方法:

在此文件夹下，运行 find . -type f -execdir sed -i ‘s/UBUNTU_PKG_NAME = “nvidia-367”/UBUNTU_PKG_NAME = “nvidia-375”/g’ ‘{}’ \;

之后make成功，编译好的文件都在bin下面。
运行了deviceQuery和bandwidthTest都没出现问题。

安装cuDNN，这个需要注册一下再安装，注意不要下载那个Power8的，那个是给power8处理器用的，我们的是amd64.
之后，解压文件，并且把相应的文件拷到相应的CUDA目录下。

sudo cp lib* /usr/local/cuda/lib64/
sudo cp cudnn.h /usr/local/cuda/include/

设置环境变量，在bashrc中添加

export CUDA_HOME=/usr/local/cuda-8.0

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

export PATH=/usr/local/cuda-8.0/bin:$PATH

安装一个库

sudo apt-get install libcupti-dev

之后安装教程采用了virtualenv安装tensorflow
注意每次使用tensorflow前需要激活virtualenv环境

source ~/tensorflow/bin/activate

不使用时，deactivate

0 0