ubuntu18.04下conda创建虚拟环境，安装tensorflow2.1报错后，成功安装tensorflow-gpu

conda创建虚拟环境，名字TF2.1（自己取）,python选择3.7版本这里

conda create -n TF2.1 python=3.7

Proceed ([y]/n)?

选择 y, 表示同意安装相关软件包。
##切换到TF2.1环境命令如下：

conda activate TF2.1

安装tensorflow2.1

pip install tensorflow==2.1

静静等待。。。。。。

测试

1 进入python 环境,在主目录直接输入：

python

2 接着导入包

import tensorflow as tf

问题：

2021-10-04 19:02:43.216801: 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; LD_LIBRARY_PATH: /usr/local/cuda-11.4/lib64:/usr/local/cuda-11.4/lib64:/usr/local/cuda-11.4/lib64　

tf.__version__

但是版本号还是输出出来了，这个应该是cuda之类的环境不一致引起的。则说明安装成功了。（我这里显示，2.1.0)

卸载

pip uninstall tensorflow

重装

pip install tensorflow-gpu

让系统自己获取版本号

结果

tf.version
‘2.6.0’