python - tensorflow 过渡到 GPU 版本

Question

我已经使用 tensorflow 一段时间了，一切正常，直到我尝试切换到 gpu 版本。

卸载了以前的tensorflow，
pip 安装 tensorflow-gpu (v2.0)
下载并安装visual studio community 2019
下载并安装 CUDA 10.1
下载并安装cuDNN

使用 CUDA 示例“deviceQuery_vs2019”进行测试并得到了积极的结果。
通过测试
英伟达 GeForce rtx 2070

使用以前的工作文件运行测试并得到错误
tensorflow.python.framework.errors_impl.InternalError: cudaGetDevice() 失败。状态:未找到 cudaGetErrorString 符号。

经过一些研究，我发现支持的 CUDA 版本是 10.0
所以我降级了版本，改变了 CUDA 路径，但没有任何改变

使用此代码

import tensorflow as tf
print("Num GPUs Available: ", 
len(tf.config.experimental.list_physical_devices('GPU')))

我明白了

2019-10-01 16:55:03.317232: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
2019-10-01 16:55:03.420537: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1618] Found device 0 with properties: 
Num GPUs Available:  1
name: GeForce RTX 2070 major: 7 minor: 5 memoryClockRate(GHz): 1.62
pciBusID: 0000:01:00.0
2019-10-01 16:55:03.421029: I tensorflow/stream_executor/platform/default/dlopen_checker_stub.cc:25] GPU libraries are statically linked, skip dlopen check.
2019-10-01 16:55:03.421849: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1746] Adding visible gpu devices: 0
[Finished in 2.01s]

CUDA 似乎可以识别卡，tensorflow 也是如此，但我无法摆脱错误:
tensorflow.python.framework.errors_impl.InternalError: cudaGetDevice() 失败。状态:未找到 cudaGetErrorString 符号。

我究竟做错了什么？我应该坚持使用cuda 10.0吗？我错过了安装的一部分吗？

Answer 1

解决了，这主要是为了避免冲突的版本炼金术。
这是我所做的(据我所知，顺序很重要)

卸载所有内容(tf、cuda、visual studio)

pip 安装 tensorflow-gpu

下载并安装 Visual Studio 社区 2017(2019 无法运行)

我还从 Visual Studio 安装了 c++ 工作负载(不确定是否有必要，但它具有所需的编译器 Visual c++ 15.x)

下载并安装 cuda 10.0(我拥有的是 10.0.130)

转到系统环境变量(在窗口栏中搜索)> 高级 > 单击环境变量...

创建新的用户变量(不要与系统变量混淆)

变量名称:CUDA_PATH，

变量值:浏览cuda目录下到版本目录(我的是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0)

该指南说您需要 cudnn 7.4.1，但我收到一个错误，提示预期版本最低为 7.6。转到 nvidia 开发人员 cudnn 存档并下载“cudnn v7.6.0 for CUDA 10.0”(确保您获得正确的文件)。解压，将cudnn文件放入对应的cuda目录(lib、include、bin)。

从那里开始，一切都像魅力一样。我无法从 Visual Studio (devicequery) 构建 cuda 示例文件，但这不是关键步骤。
几乎每个错误都是由于文件版本不兼容造成的，我花了 3-4 天的时间来找出正确的组合。希望有所帮助:)