在安装tensorflow之前,首先列出需要的配置环境:Ubuntu16.04lts系统,支持GPU运算的显卡(一般为NVDIA的显卡),Cuda8.0,Cudnn v5.1,Anaconda Python2.7(或者3.5)的环境。
1.安装Cuda
首先下载Cuda8.0 (https://developer.nvidia.com/cuda-downloads)
然后进入下载目录,执行以下命令,就可安装Cuda
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
安装完成后配置环境变量,在home下的 .bashrc中加入:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr