Блог

Я работаю в виртуальной среде.

Таким образом, в основном я установил тензорный поток, и когда я запускаю import tensorflow as tf вывод, который я получаю, это.

 2021-08-11 17:42:31.395050: 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-08-11 17:42:31.395365: 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.
 

Я попытался установить CUDA и tf-nightly-gpu , как предлагали некоторые исключения, но это не устранило проблему.

текущий список пакетов

 Package                 Version
----------------------- -------------------
absl-py                 0.13.0
astunparse              1.6.3
cachetools              4.2.2
certifi                 2021.5.30
charset-normalizer      2.0.4
click                   8.0.1
colorama                0.4.4
cuda                    0.0.1
filelock                3.0.12
flatbuffers             1.12
gast                    0.4.0
gensim                  4.0.1
google-auth             1.34.0
google-auth-oauthlib    0.4.5
google-pasta            0.2.0
grpcio                  1.39.0
h5py                    3.1.0
huggingface-hub         0.0.12
idna                    3.2
joblib                  1.0.1
keras-nightly           2.7.0.dev2021081100
Keras-Preprocessing     1.1.2
libclang                11.1.0
Markdown                3.3.4
nltk                    3.6.2
numpy                   1.19.5
oauthlib                3.1.1
opt-einsum              3.3.0
packaging               21.0
Pillow                  8.3.1
pip                     21.2.3
protobuf                3.17.3
pyasn1                  0.4.8
pyasn1-modules          0.2.8
pyparsing               2.4.7
PyYAML                  5.4.1
regex                   2021.8.3
requests                2.26.0
requests-oauthlib       1.3.0
rsa                     4.7.2
sacremoses              0.0.45
scikit-learn            0.24.2
scipy                   1.7.1
sentence-transformers   2.0.0
sentencepiece           0.1.96
setuptools              57.4.0
six                     1.15.0
smart-open              5.1.0
tb-nightly              2.6.0a20210806
tensorboard             2.6.0
tensorboard-data-server 0.6.1
tensorboard-plugin-wit  1.8.0
tensorflow              2.5.1
tensorflow-estimator    2.5.0
tensorflow-hub          0.12.0
termcolor               1.1.0
tf-estimator-nightly    2.7.0.dev2021081101
threadpoolctl           2.2.0
tokenizers              0.10.3
torch                   1.9.0
torchvision             0.10.0
tqdm                    4.62.0
transformers            4.9.2
typing                  3.7.4.3
typing-extensions       3.7.4.3
urllib3                 1.26.6
Werkzeug                2.0.1
wheel                   0.37.0
wrapt                   1.12.1
 

Как настроить графический процессор, чтобы tensorflow использовал графический процессор? Я понимаю, что в нем говорится, что вы можете включить использование графического процессора, но нет объяснения, как и куда идти в первую очередь.

Спасибо

Я полагаю, вы используете Windows (из-за сообщения об ошибке)

Вам необходимо установить NVIDIA CUDA — tensorflow 2.5.x использует 11.2 (см. Таблицу версий) также я бы рекомендовал установить CUDNN 8.x, так как некоторые слои (например, LSTM) будут использовать его

Проверьте эти руководства:

Руководство по установке Tensorflow 2

Ссылка для скачивания CUDA

Ссылка для скачивания CUDNN

Учебное пособие — подробное руководство для более старой версии CUDA, но если вы используете более новые версии CUDA/CUDNN, worflow в основном такой же