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

RTX3080 安装tensorflow-gpu and cudnn

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

RTX3080 安装tensorflow-gpu and cudnn

今天要测试一个tensorflow的代码 于是就尝试安装了一下在liunx服务器上 1.我的原始环境是torch1.8.0+cu111 ,初始安装的是tensorflow-gpu==2.4.0
import tensorflow as tf
print('GPU',tf.test.is_gpu_available())

结果输出为False 百度之后发现是cudn和cudnn的版本太低
因为我的conda是换过源的 ,所以输入

conda search xxx # 在当前所有通道查找可以安装的xxx

找到你想用的版本并安装

conda install cudnn/cudatoolkit

测试输出还是False
安装 2.5.0版本的gpu 输出结果

2. 在运行代码的时候又出现 AttributeError: module ‘keras.utils.generic_utils’ has no attribute 'populate_dict_with_module_objec的问题

卸载之后的keras==2.3.0 重新安装
代码可以进行测试 虽然效果不是很好
最后说一句 怪不得现在大部分都用pytorch 就是版本和环境的对应问题就要弄好久

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

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号