我的系统及环境:
Ubuntu14.04(64位), Python2.7, Nvidia GeForce GTX1050TI
Tensorflow GPU 版本要求显卡的Compute Capability >= 3.0, GTX1050TI的Compute Capability = 6.1 (NVIDIA显卡计算能力查看网址 https://developer.nvidia.com/cuda-gpus), 因此,我装的是Tensorflow GPU版本。
安装TF GPU版本,必要要先装CUDA和cuDNN。
1.检查环境
lspci | grep -i nvidia (查看显卡型号)
uname -m && cat /etc/*release
gcc --version
uname -r