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’

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐