主页 / 问题 / 1591945
Accepted
Alice Magic
Asked:2024-08-26 20:22:56 +0000 UTC2024-08-26 20:22:56 +0000 UTC

我安装了 CUDA，但 PyCharm Community Edition 上的 Python 没有看到它

  • 772

安装cuda_11.2.0_460.89_win10.exe并cudnn_8.1.0_windows.exe。

安装了 CUDAC:\Program Files\NVIDIA GPU Computing Toolkit

安装最新的驱动程序：GeForce Game Ready- 版本：560.94

环境变量路径：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\libnvvp

在命令行上，命令：nvcc --version产生：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Nov_30_19:15:10_Pacific_Standard_Time_2020
Cuda compilation tools, release 11.2, V11.2.67
Build cuda_11.2.r11.2/compiler.29373293_0

在 PyCharm Community Edition 2024.1.3 中，在终端中通过pip list-tensorflow 2.17.0

但回应该节目：

import tensorflow as tf

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

写道：

可用 GPU 数量：0

为什么？我做错了什么？

python

1 个回答

  1. Best Answer

    tensorflow 2.17.0 - 对于 CUDA v11.2 来说太晚了版本低于兼容性 https://www.tensorflow.org/install/source_windows?hl=ru#gpu 对于tensorflow 2.17.0，您需要 cuDNN 8.9 CUDA - 12.3 来实现 CUDA \ v11.2 您可以安装从 2.5 到 2.11 的tensorflow 您可能还需要更新 Visual C 和 dirx。我只设置好了2.5

    • 2

相关问题