如何设置 TensorFlow 以与 GPU 一起使用

问题描述 投票:0回答:1

我想安装支持我的 GPU 的 TensorFlow。我有一块 GeForce GTX 4070 GPU。它的计算能力为8.9,绰绰有余。 我尝试使用以下命令安装张量流: pip install tensorflow[和-cuda]

但是当我验证 GPU 支持时,列表为空: python3 -c“将tensorflow导入为tf;打印(tf.config.list_physical_devices('GPU'))”

我已安装 NVIDIA® GPU 驱动程序、CUDA® Toolkit 11.8 和 cuDNN SDK 8.6.0。 请问有知道这个问题的人可以帮助我吗?

我使用的是 Windows 10。 除了我的问题之外,是否可以在 Windows 10 上获得 GPU 支持?

我每次都得到以下错误信息:

2024-03-09 21:29:34.106076:Wtensorflow/stream_executor/platform/default/dso_loader.cc:64]无法加载动态库“cudart64_110.dll”; dlerror:找不到 cudart64_110.dll 2024-03-09 21:29:34.106172:我tensorflow/stream_executor/cuda/cudart_stub.cc:29]如果您的计算机上没有设置GPU,请忽略上面的cudart dlerror。 2024-03-09 21:29:34.986368:W tensorflow/stream_executor/platform/default/dso_loader.cc:64] 无法加载动态库“cudart64_110.dll”; dlerror:找不到 cudart64_110.dll 2024-03-09 21:29:34.986691:W tensorflow/stream_executor/platform/default/dso_loader.cc:64] 无法加载动态库“cublas64_11.dll”; dlerror:找不到 cublas64_11.dll 2024-03-09 21:29:34.986909:W tensorflow/stream_executor/platform/default/dso_loader.cc:64] 无法加载动态库“cublasLt64_11.dll”; dlerror:找不到 cublasLt64_11.dll 2024-03-09 21:29:34.987114:W tensorflow/stream_executor/platform/default/dso_loader.cc:64] 无法加载动态库“cufft64_10.dll”; dlerror:找不到 cufft64_10.dll 2024-03-09 21:29:34.987306:W tensorflow/stream_executor/platform/default/dso_loader.cc:64] 无法加载动态库“curand64_10.dll”; dlerror:找不到 curand64_10.dll 2024-03-09 21:29:34.987496:W tensorflow/stream_executor/platform/default/dso_loader.cc:64] 无法加载动态库“cusolver64_11.dll”; dlerror:找不到 cusolver64_11.dll 2024-03-09 21:29:34.987682:W tensorflow/stream_executor/platform/default/dso_loader.cc:64] 无法加载动态库“cusparse64_11.dll”; dlerror:找不到 cusparse64_11.dll 2024-03-09 21:29:34.987872:W tensorflow/stream_executor/platform/default/dso_loader.cc:64] 无法加载动态库“cudnn64_8.dll”; dlerror:找不到 cudnn64_8.dll 2024-03-09 21:29:34.987894:W tensorflow/core/common_runtime/gpu/gpu_device.cc:1934] 无法 dlopen 某些 GPU 库。如果您想使用 GPU,请确保正确安装上述缺少的库。请按照 https://www.tensorflow.org/install/gpu 上的指南了解如何下载和设置您的平台所需的库。 正在跳过注册 GPU 设备... []

tensorflow installation deep-learning gpu cudnn
1个回答
0
投票

尝试在您的环境中使用 conda 安装 Cuda 和 Cudnn:

conda install cudnn

不要 pip 安装 cuda。先删除它,然后尝试通过 conda 安装。

© www.soinside.com 2019 - 2024. All rights reserved.