项目场景:
深度学习框架:Tensorflow2.4.1
CUDA:11.1
显卡型号:2080Ti
问题描述:
我根据驱动型号,安好了CUDA,并且添加了cuDNN成功:
pip 安装Tensorflow:
pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
安装Tensorflow后显示有错,无法使用GPU:
import tensorflow as tf # 导入tensorflow
tf.test.is_gpu_available() # 验证是否tensorflow GPU可用
最后一行显示:
Could not load dynamic library ‘cusolver64_10.dll‘; dlerror: cusolver64_10.dll not found
原因分析:
根据显示已经可以确认有个文件cusolver64_10.dll
没有找到!
解决方案:
1、下载对应所缺失的 xx.dll文件
2、进入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin的目录中,将对应的cusolver64_10.dll文件粘贴到此目录下
再次运行代码,成功:
import tensorflow as tf
gpu_ok = tf.test.is_gpu_available()
print("tf version",tf.__version__)
print("use GPU",gpu_ok)