栏目分类:
子分类:
返回
终身学习网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
终身学习网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间:发布时间: 百科书网 趣学号
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’

转载请注明:文章转载自 www.051e.com
本文地址:http://www.051e.com/it/294272.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 ©2023-2025 051e.com

ICP备案号:京ICP备12030808号