我表示ubuntu经常出问题,已经安装了n次tensorflow了,所以在这里记下最简单最快的方法,大家安装的时候按照流程跑就可以了,有什么问题可以在评论里问我~
1.下载cuda和cudnn
https://developer.nvidia.com/cuda-downloads (选择linux-x86_64-ubuntu-15.04-deb(local))
https://developer.nvidia.com/cudnn (Cudnn V4)
2. 安装cuda
cd ~/Downloads(进入下载的目录),执行sudo dpkg -i 你的下载包
执行sudo apt-get update
执行sudo apt-get install cuda
执行sudo gedit ~/.bashrc,在最后面,64位的话粘贴以下内容:
export PATH=/usr/local/cuda-7.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
3. 把cudnn里的文件复制到你安装cuda的地方,一般是在 /usr/local/cuda7.5 或者 /usr/local/cuda (两个都试一下)
sudo cp cuda/libcudnn* /usr/local/cuda-7.5/lib64
sudo cp cuda/cudnn.h /usr/local/cuda-7.5/include
4.安装tensorflow,这里使用pip安装的方法
sudo apt-get install python-numpy python-scipy python-dev python-pip python-nose g++ libopenblas-dev git
export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.9.0rc0-cp27-none-linux_x86_64.whl
sudo pip install --upgrade $TF_BINARY_URL
注:如果之前已经装了,要先卸载,用sudo pip uninstall tensorflow