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

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

版本对应关系
ubuntu tensorflow1.1x版本安装及常见问题_第1张图片

有时候,我们需要在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

你可能感兴趣的:(Linux,tensorflow,cuda,anaconda)