解决tensorflow2.4.0版本出现failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED错误的方法

企业开发 2023-09-09 20:53:43 阅读次数: 0

在主代码中加入

physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

即可解决这个错误

相当简单

转载自blog.csdn.net/qq_19313495/article/details/114114114

今日推荐

美国拟限制 AI 大模型出口中国和俄罗斯

openKylin 社区生态委员会第六次会议圆满召开

阿里云正式发布通义千问 2.5

Pop!_OS 的 COSMIC 桌面完成 App Store 上架工作

报告：Django 仍然是 74% 开发者的首选

TIOBE 5 月榜单：Fortran “复活”进入 Top 10

周排行

BPM为企业带来的实际利益

好程序员web前端分享css常用属性缩写

Java文件下载（excel）

css样式的动态添加及显示和隐藏等零碎用法

axios全局配置以及拦截器

使用Logstash来实时同步MySQL和log日志数据到ES

C++获取当前时间（年月日、时分秒、毫秒）

01 官网下载各种CentOS教程（超详细版）

每日归档

2024-05-14(0)

2024-05-13(18)

2024-05-12(0)

2024-05-11(38)

2024-05-10(38)

2024-05-09(35)

2024-05-08(42)

2024-05-07(14)

2024-05-06(40)

2024-05-05(0)