ubuntu tensorflow1.1x版本安装及常见问题

Posted 2022-12-09 栋次大次

hai，tensorflow在安装过程中，需要注意版本对应关系。比如tensorflow-gpu1.12对应cuda9.0和cudnn。

版本对应关系

有时候，我们需要在cuda安装的版本较高的环境中安装较低版本的tensorflow。笔者最近在cuda10.1的机器上安装tensorflow1.12，分享下安装过程。

# 安装1.12版本，1.12版本对应的是cuda9 和 cudnn7
conda install cudatoolkit=9.0
conda install cudnn=7
pip install tensorflow-gpu==1.12.0

测试：

import tensorflow as tf
print(tf.test.is_gpu_available())

安装其它版本的tensorflow也是类似过程，先安装对应的cudatoolkit，再安装对应的cudnn，最后安装tensorflow-gpu。

常见问题

假设安装成功，并且安装的是tensorflow1.1x版本，在运行程序时出现以下错误或者警告：

/home/kitty/anaconda3/envs/py36/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:526: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint8 = np.dtype([("qint8", np.int8, 1)])

这是因为，numpy版本过高，可以用下面命令解决。

pip uninstall numpy
pip install numpy==1.16.4

备注：对于1.15及更早版本，CPU和GPU软件包是分开的：

pip install tensorflow==1.15 # CPU
pip install tensorflow-gpu==1.15 # GPU