#python #tensorflow #gpu #amd
Вопрос:
Я новичок в tensorflow, пытаюсь установить tensorflow на своем ноутбуке Windows и настроить встроенный 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.
Заранее спасибо!
Ответ №1:
Следуйте этим инструкциям.
«Поддержка графического процессора TensorFlow требует широкого спектра драйверов и библиотек. Чтобы упростить установку и избежать конфликтов библиотек, мы рекомендуем использовать образ докера TensorFlow с поддержкой графического процессора (только для Linux). Для этой настройки требуются только драйверы графического процессора NVIDIA®».
— Вы не сможете сделать это с помощью простого pip . Вам нужны драйверы графического процессора Nvidia, набор инструментов CUDO, CUPTI и cuDNN SDK