这段时间在搞深度学习,代码在别人的电脑上的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,问题解决