在Python中使用TensorflowGPU加载DLL失败

2024-04-28 18:18:38 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用Windows7 X64,并试图为tensorflow建立一个gpu环境。我下载了CUDA Toolkit 10.1、NVIDIA驱动程序和CuDNN,并安装了它们。然而,根据tensorflow的官方网站,我在安装后找不到目录\tools\cuda\bin。所以我做了一个,把cudnn64_7.dll放进去。然后我在系统的PATH中添加了所有必需的路径

然后,我使用Anaconda的命令窗口创建环境:

conda create -n tensorflow_gpu pip python

输入后,我安装了tensorflow-gpu如下:

pip install --ignore-installed --upgrade tensorflow-gpu

有人告诉我这是成功的。通过查看conda列表,它就在那里。但是,当我使用以下代码进行测试时:

import tensorflow as tf

出现了以下情况: Screenshot of traceback

我试图重新开始并重新安装NVIDIA的东西,但问题依然存在。发生了什么事?我如何修复


Tags: pip目录gpu环境tensorflow驱动程序toolstoolkit