tensorflow 使用RTX 3070笔记本GPU在Windows系统下进行CNN模型训练

yr9zkbsy  于 2023-02-24  发布在  Windows
关注(0)|答案(1)|浏览(342)

我尝试使用我的笔记本电脑RTX 3070 GPU的CNN模型训练,因为我必须采用exhastive网格搜索来调整超参数。我尝试了许多不同的方法,但是,我不能得到它做。谁能好心地指出我在正确的方向?
我遵循了以下程序。程序:
1.已安装NVIDIA CUDA工具包11.2
1.通过下载文件(bin、include、lib)并将其粘贴到NVIDIA GPU计算工具包/CUDA/V11.2中,安装了NVIDIA cuDNN 8.1
1.通过在bin和libnvvm的系统路径中包含路径来设置环境变量。
1.在新的conda环境中安装了tensorflow 2.11和python 3.8。
然而,我无法设置系统使用可用的GPU。代码似乎只使用CPU,当我查询以下请求时,我得到以下输出。
查询:

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

输出:

TensorFlow version: 2.11.0
Num GPUs Available:  0

我错过了什么吗?或者有人和我有同样的问题?

umuewwlo

umuewwlo1#

您应该使用DirectML插件。* 从tensorflow 2.11 GPU支持已下降到本机窗口。* 您需要使用DirectML插件。您可以按照教程here安装

相关问题