如何在tensorflow上使用gpu

时间:2020-07-17 23:55:13

标签: python tensorflow cudnn

所以我试图在GPU上运行python代码以使用预先训练的机器学习模型 我有以下初始化: CUDA 9.0

cudnn-7.6.5

python 3.5.2

tensorflow 1.5

运行代码时,出现以下错误:

2020-07-18 02:46:57.624022:EC:\ tf_jenkins \ workspace \ rel-win \ M \ windows-gpu \ PY \ 35 \ tensorflow \ stream_executor \ cuda \ cuda_dnn.cc:385]无法创建cudnn句柄:CUDNN_STATUS_NOT_INITIALIZED 2020-07-18 02:46:57.629159:EC:\ tf_jenkins \ workspace \ rel-win \ M \ windows-gpu \ PY \ 35 \ tensorflow \ stream_executor \ cuda \ cuda_dnn.cc:389]错误检索驱动程序版本:未实现:内核报告的驱动程序版本未在Windows上实现 2020-07-18 02:46:57.633213:FC:\ tf_jenkins \ workspace \ rel-win \ M \ windows-gpu \ PY \ 35 \ tensorflow \ core \ kernels \ conv_ops.cc:717]检查失败:流-> parent()-> GetConvolveAlgorithms(conv_parameters.ShouldIncludeWinogradNonfusedAlgo(),&algorithms) 2020-07-18 02:46:57.633864:

提到“无法创建cudnn句柄:CUDNN_STATUS_NOT_INITIALIZED”

因此,请您考虑如何解决此问题...如果需要进一步澄清问题,请提前通知我。

0 个答案:

没有答案