张量流2.1.0的安装错误

时间:2020-01-28 17:38:44

标签: python tensorflow tensorflow2.0

  1. t1.doSomething(new Test2()); == pip3 --version
  2. pip 20.0.2 from /home/nitin/anaconda3/envs/tensorflow/lib/python3.7/site-packages/pip (python 3.7) == python --version

我用python 3.7.6创建了一个环境。我在其中安装了tensorflow 2.0,我使用了conda。我在张量流环境中使用conda create --name tensorflow对其进行了升级

现在,当我做pip install --upgrade tensorflow时,出现以下错误

import tensorflow as tf

我似乎不明白出了什么问题。有帮助吗?

2 个答案:

答案 0 :(得分:2)

这只是来自tensorflow的警告,指出如果您想提高某些模型的延迟和吞吐量,则可以利用tensorRT的功能。但是,由于未安装libnvinfer,因此无法使用它。 为了安装它,

wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb

sudo apt install ./nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb

sudo apt-get update

sudo apt-get install -y --no-install-recommends \
    libnvinfer6=6.0.1-1+cuda10.1 \
    libnvinfer-dev=6.0.1-1+cuda10.1 \
    libnvinfer-plugin6=6.0.1-1+cuda10.1

这是参考tensorflow

提供的安装说明

答案 1 :(得分:1)

如果您使用的是python 3.7 CPU,请按照以下方式进行升级:pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.1.0-cp37-cp37m-manylinux2010_x86_64.whl