最近换了个带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
以上就是我安装过程中遇到的问题及其解决方案啦,如果帮到你的话,点个赞再走吧~