一定要选和自己CUDA和cudnn版本兼容的
安装参考1https://blog.csdn.net/weixin_41368740/article/details/102254423
参考2https://blog.csdn.net/weixin_38283428/article/details/84201733
最好一开始就安装tensorflow-gpu,我不知道CPU和GPU版能不能共存,当我安装好GPU版时,运行出现了错误,找了很多方法都不行,然后把GPU版卸了,卸了发现CPU版也不能用了,查询发现说,可能卸载GPU版时把CPU版需要的包也卸了,然后把CPU版也卸了,直接安装了GPU版,安装GPU版后,运行出现了UnknownError (see above for traceback): Failed to get convolution algorithm.错误,可以参考https://blog.csdn.net/qq_41621362/article/details/89321361
顺利解决
tensorflow-gpu 安装
最新推荐文章于 2023-11-01 11:30:37 发布