昨天,我的tensorflow突然报了一个矩阵计算的错误,但是我用的是keras的高级API,这时我一下子想到了版本兼容的问题上去。
发现自己的cuda版本为12.1,而我之前的cuda是11.2,果然不兼容,应该是cuda自动更新所导致。
重新安装了cuda,在conda中下载了cudnn还有cudatoolkit,这时没有矩阵计算的错误了,而是提示无法加载某个dll文件,无法调用GPU。
Could not load dynamic library
skipping GPU register
我来到cuda的文件夹下搜索这个dll文件,不出意外,没有找到这个dll文件。
随后,去官网手动下载了cudnn,然后把cudnn的那几个文件夹复制到了cuda的文件夹下,然后再从cuda文件夹下搜索这个dll文件,终于发现了该dll文件。
重启Pycharm,成功调用GPU