1. 安装依赖项
sudo apt-get install openjdk-8-jdk git python-dev python3-dev python-numpy python3-numpy python-six python3-six build-essential python-pip python3-pip python-virtualenv swig python-wheel python3-wheel libcurl3-dev libcupti-dev
2.安装pip
sudo apt-get install python-pip python-dev
3.安装tensorflow
sudo pip install tensorflow-gpu
4.测试tensorflow是否安装成功
python
import tensorflow as tf
出现错误:
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
因为自动安装的tensorflow1.5版本,支持cuda9.0,不支持8.0,所以应该安装其他本版的tensorflow
重新安装
sudo pip install tensorflow-gpu==1.2
再次测试tensorflow
python
import tensorflow as tf
出现新的错误:
ImportError: libcudnn.so.5: cannot open shared object file: No such file or directory
原因是我的使用cudnn6.0版本的,而tensorflow1.2支持cudnn5.0版本的,因此需要重新安装适用于cudnn6.0版本的tensorflow
重新安装,输入:
sudo pip install tensorflow-gpu==1.3
安装测试成功。