如何使用GPU与tensorflow



我在虚拟环境中运行。

基本上我已经安装了张量流,当我运行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.

我已经尝试安装CUDAtf-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

我如何设置GPU使tensorflow使用GPU。我知道它说你可以启用GPU使用,但是没有解释如何或从哪里开始。

感谢

我想你使用Windows(由于错误消息)

你必须安装NVIDIA CUDA - tensorflow 2.5。x使用11.2(见版本表)此外,我建议安装CUDNN 8。因为有些层(例如LSTM)会使用它

查看这些指南:

Tensorflow 2安装指南

CUDA下载链接

CUDNN下载链接

教程-一个详细的教程为一个较旧的CUDA版本,但如果你使用较新的CUDA/CUDNN版本,工作流程基本上是相同的

相关内容

  • 没有找到相关文章

最新更新