当你遇到failed to create cublasLt handle: CUBLAS_STATUS_INTERNAL_ERROR

作者分享了解决TensorFlow 2.0在NVIDIA 3060 GPU上运行问题的方法,包括检查版本兼容性和通过设置CUDA_VISIBLE_DEVICES环境变量。建议在遇到类似问题时参考此经验。
摘要由CSDN通过智能技术生成

这段时间在搞深度学习,代码在别人的电脑上的tensorflow1.0版本上可以正常运行

但是自己的3060不知道为啥就是不行,一查好像是3060不支持CDUA10.0以下版本

那咱就装2.0呗,装了之后一直报这个错误,查了一万种方法都不知道咋解决

今天终于解决了,分享出来让大家少浪费一点时间

 遇到这个问题有两种解法

1.看看你的tensorflow版本和cuda以及cudnn版本是否吻合

我是3060+tensorflow-gpu2.4.1+cuda11.1+cudnn8.04/8.05

2.确认都对了以后,在代码最上方加入

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "1"

OK,问题解决

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值