当先锋百科网

首页 1 2 3 4 5 6 7

一,首先显卡更新为nvidia

1,先程序搜索更新

$ sudo apt-get update

2,去设置——软件和更新中的附加驱动选择nvidia驱动

二,安装CUDA

$ sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

三,安装cudnn

$ tar xvzf cudnn-8.0-linux-x64-v6.0.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 /usr/local/cuda/lib64/libcudnn*

五,配置环境变量

$ sudo gedit ~/.bashrc

加入以下内容

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

命令行中执行

source ~/.bashrc

六,pip安装tensorflow_gpu_1.4.0

$ sudo pip3 install tensorflow_gpu-1.4.0-cp34-cp34m-manylinux1_x86_64.whl 

七,测试安装的tensorflow

至此tensorflow 安装教程结束