如何在windows中为tensorflow设置gpu(amd radeon r5 m330)

ac1kyiln  于 2021-09-29  发布在  Java
关注(0)|答案(1)|浏览(580)

我是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.

提前谢谢!

bd1hkmkf

bd1hkmkf1#

按照这些说明操作。
“tensorflow gpu支持需要各种驱动程序和库。为了简化安装并避免库冲突,我们建议使用支持gpu的tensorflow docker映像(仅限linux)。此设置只需要NVIDIA®GPU驱动程序。
--你不能用简单的 pip . 您需要nvidia gpu驱动程序、cudo工具包、cupti和cudnn sdk

相关问题