作为一个初学者在装Tensorflow时遇到很多问题,填了几个坑之后,找到一个比较简单的办法,所以写出来作为参考。
利用pip安装方法
安装显卡官方驱动
1.点桌面左上角关机图标,找到“系统设置”->软件和更新->附加驱动
2.在“附加驱动”里,系统会自动搜索驱动,列表里会提供对应你显卡的最新版官方驱动。
3.最后点“应用更改”,等待安装完毕即可。
安装CUDA【Debian安装】
1、下载安装 (https://developer.nvidia.com/cuda-downloads)
进入下载文件所在目录,执行下列命令:
$ sudo dpkg --install cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda
安装cuDNN
1、下载安装Cudnn v5.1(https://developer.nvidia.com/cudnn)
进入下载文件所在目录,执行下列命令:
$ tar xvzf cudnn-8.0-linux-x64-v5.1.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*
2、配置环境变量:
在terminal根目录中输入以下命令:
$ sudo gedit ~/.bash_profile
然后在打开的文本末尾加入:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
继续在terminal中输入:
$ source ~/.bash_profile
安装pip
$ sudo apt-get install python-pip python-dev
$ sudo apt-get install python-numpy swig python-dev python-wheel
安装TensorFlow
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl
$ sudo -H pip install --upgrade $TF_BINARY_URL
详细可见:Tensorflow