TensorFlow 2.1从终端加载时返回警告,从Jupyter加载时返回错误

2024-09-27 23:28:01 发布

您现在位置:Python中文网/ 问答频道 /正文

我已经在家用笔记本电脑(Ubuntu 18.04)的virtualenv中安装了tensorflow 2.1.0。当Iimport tensorflow as tf从终端收到以下警告:

2020-03-12 12:17:56.485098: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer.so.6'; dlerror: libnvinfer.so.6: cannot open shared object file: No such file or directory

2020-03-12 12:17:56.485179: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer_plugin.so.6'; dlerror: libnvinfer_plugin.so.6: cannot open shared object file: No such file or directory

2020-03-12 12:17:56.485189: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:30] Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.

但是,导入工作正常,我可以像往常一样运行代码

但当我试图在Jupyter笔记本上做同样的事情时,我得到了错误import tensorflow as tf返回:

ModuleNotFoundError: No module named 'tensorflow'

我的笔记本电脑上没有GPU,但tensorflow 2.1应该能够在GPU和CPU专用机器上运行。事实上,我在工作中成功地安装了它,没有任何问题。我能做些什么来解决这个问题


Tags: nodefaultstreamgputftensorflowasloader
1条回答
网友
1楼 · 发布于 2024-09-27 23:28:01

您可以使用以下代码抑制警告消息

import logging, os
logging.disable(logging.WARNING)
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
import tensorflow as tf

关于Jupyter笔记本上的错误

Open anaconda-navigator and create new environment and launch the jupyter-notebook. Now install tensorflow

!pip install tensorflow

相关问题 更多 >

    热门问题