时间戳

首页 > 解决方案 > 如何在 Windows 中为 Tensorflow 设置 GPU (AMD Radeon R5 M330)

python - 如何在 Windows 中为 Tensorflow 设置 GPU (AMD Radeon R5 M330)

问题描述

我是 tensorflow 的新手，尝试在我的 Windows 笔记本电脑中安装 tensorflow 并配置内置的 AMD Radeon R5 M330，任何指南/步骤都会非常有帮助。

到目前为止，我已经安装了以下内容

pip install tensorflow
pip install tensorflow-gpu
pip install tf-nightly-gpu

导入 tf 时收到以下警告

>>>import tensorflow as tf
2021-07-27 14:11:41.339706: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2021-07-27 14:11:41.339901: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

提前致谢 ！

标签： pythontensorflowgpuamd

解决方案

遵循这些说明。

“TensorFlow GPU 支持需要各种驱动程序和库。为了简化安装并避免库冲突，我们建议使用支持 GPU 的 TensorFlow Docker 映像（仅限 Linux）。此设置仅需要 NVIDIA® GPU 驱动程序。”

- 你将无法使用简单的pip. 您需要 Nvidia GPU 驱动程序、CUDO 工具包、CUPTI 和 cuDNN SDK

推荐阅读