操作系统:Ubuntu 16.04
TF官方说明
https://www.tensorflow.org/versions/r1.1/install/install_linux
明确需要安装:CUDA Toolkit8.0 , cuDNN v5.1 , libcupti-dev
通过以下指令可安装libcupti-dev
$ sudo apt-get install libcupti-dev
CUDA和cuDNN的安装参考NVIDIA’s documentation
系统要求:
操作步骤:
1)Verify you have a CUDA-Capable GPU
$ lspci | grep -i nvidia
终端内输入以上指令可检查
2)Verify you have a supported version of Linux
$ uname -m && cat /etc/*release
终端内输入以上指令可检查
3)Verify the system has gcc installed
$ gcc --version
终端内输入以上指令可检查
4)Verify the System has the Correct Kernel Headers and Development Packages Installed
$ sudo apt-get install linux-headers-$(uname -r)
终端内可通过以上指令安装/检查
5)Download the NVIDIA CUDA Toolkit
NVIDIA CUDA Toolkit 获取网址: http://developer.nvidia.com/cuda-downloads.
选择使用平台。
下载链接:上述版本
下载后终端指令安装。
发现自己的GPU不满足要求。。。
TF要求GPU的Compute Capability(点击查看)至少为3.0,我的没到3.0,疏忽了
等换了电脑再搭吧
补充待续…