安装tensorflow-gpu 2.9.0报错及解决

  • Post author:
  • Post category:其他


最近换了个带GPU的新装备,兴致勃勃下载anaconda和pycharm,想着安装tensorflow-gpu不就新创个环境,然后

pip install tensorflow-gpu==2.9.0

,之后用gpu跑代码就像呼吸一样简单!

然而实际过程中遇到了各种各样的问题,网上现有的完整教程基本都是21、22年的,anaconda、pycharm、cuda、cudnn都有更新,下面就来分享一下我这次遇到的问题和解决方法吧


1、一定要注意cuda和cudnn版本对应关系


这个提前要查好,网站链接:

https://tensorflow.google.cn/install/source_windows?hl=zh-cn#gpu


最新版本对应如下:

在这里插入图片描述


2、安装并添加cuda和CuDnn的环境变量后,检验是否安装成功


方法如下:cmd进入到cuda的安装路径

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\demo_suite

首先执行

deviceQuery.exe

,若出现以下界面:

在这里插入图片描述

再输入

bandwidthTest.exe

,若出现以下界面:

在这里插入图片描述

出现上述两次

Result = Pass

则安装成功,假如有报错那大概率是NVIDAI Driver没有安装,去这个链接:

https://www.nvidia.cn/Download/index.aspx?lang=cn

,根据自己电脑的显卡驱动选择相应NVIDAI驱动进行安装,显卡驱动到设置 –> 设备管理器 –> 展开显示适配器 –> GPU来看。

应该在下面这行


3、Pycharm添加解释器环境出错


如果你已经成功安装tensorflow-gpu,这里插一句,检测是否成功安装可打开cmd输入语句:

>>> import tensorflow as tf
>>> tf.test.is_gpu_available()

若打印True,则安装成功,如下图:

在这里插入图片描述

但在pycharm导入环境时出现了这样的错误:

在这里插入图片描述

由于我的pycharm是最新版,所以并不像网上一些教程指出的是上述报错原因是没有安装PythonOCC,而是绑定错了。不应该绑定python.exe,而应该是conda.exe,路径在

\anaconda3\Scripts\conda.exe

以上就是我安装过程中遇到的问题及其解决方案啦,如果帮到你的话,点个赞再走吧~



版权声明:本文为LUSIA_TT原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。