TensorFlow无法检测CUDA

dly7yett 于 2023-02-13 发布在其他

关注(0)|答案(1)|浏览(175)

我安装了CUDA工具包和CUDNN，并将其添加到PATH中，但此函数仍返回False。

import tensorflow as tf
print(tf.test.is_built_with_cuda())

我使用Anaconda发行版中的JupyterNotebook。
操作系统是Windows 11。CUDA v11。CUDNN v8.7。也有zlib dll。
该命令
nvcc -V
在PowerShell工作中，它输出CUDA版本。
由于CUDA，TensorFlow也无法检测显卡。

print(tensorflow.config.list_physical_devices())

此代码仅返回有关CPU的信息。

tensorflow

来源：https://stackoverflow.com/questions/75288243/tensorflow-does-not-detect-cuda

1条答案

按热度按时间

twh00eeo1#

您需要按照此构建配置安装CUDA 11.2和cuDNN 8.1，以便在系统中启用Tensorflow GPU支持。
请安装GPU支持所需的所有软件，并将这些软件的路径设置为bin目录。
然后在conda环境中使用以下代码进行TF-gpu设置。

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
pip install --upgrade pip
# Anything above 2.10 is not supported on the GPU on Windows Native
pip install "tensorflow-gpu<2.11"

要验证GPU设置：

python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

赞(0）回复(0）举报 2023-02-13

我来回答

TensorFlow无法检测CUDA

1条答案

相关问题

热门标签

最新问答