Как узнать, какой TensorFlow установлен в моей системе Windows? Будь то процессор или графический процессор TensorFlow

Я опробовал следующую команду в командной строке Anaconda,

pip show tensorflow

что дало мне результат,

Name: tensorflow
Version: 1.10.0
Summary: TensorFlow is an open source machine learning framework for everyone.
Home-page: https://www.tensorflow.org/
Author: Google Inc.
Author-email: opensource@google.com
License: Apache 2.0
Location: d:\softwares\pycond\envs\tensorflow2\lib\site-packages
Requires: termcolor, absl-py, grpcio, tensorboard, wheel, protobuf, gast, six, numpy, setuptools, astor

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

Источник

1 ответ

Пакет pip TensorFlow включает поддержку графического процессора для карт с поддержкой CUDA®.

      pip show tensorflow

Для более старых версий TensorFlow:

Для версий 1.15 и более ранних пакеты ЦП и ГП разделены:

      pip install tensorflow==1.15      # CPU
pip install tensorflow-gpu==1.15  # GPU

Итак, имена пакетов для версий 1.15 и старше отличаются.

Источник
Другие вопросы по тегам