Блог

вот код, который я использую, чтобы проверить, работает ли tf.gpu или нет

 import tensorflow as tf 
if tf.test.gpu_device_name(): 
    print('Default GPU Device:{}'.format(tf.test.gpu_device_name()))
else:
   print("Please install GPU version of TF")
 

и вот ошибка

 Failed to load the native TensorFlow runtime.
See https://www.tensorflow.org/install/errors
for some common reasons and solutions.  Include the entire stack trace
above this error message when asking for help.
2020-11-22 21:53:40.971514: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
2020-11-22 21:53:40.971756: 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.
 

Чтобы использовать графический процессор с Tensorflow, необходимо установить графическую версию Tensorflow

python -m pip install tensorflow-gpu

Убедитесь, что вы также используете 64-разрядную версию python, поскольку она будет работать только с этими параметрами.

Редактировать:

Начиная с версии Tensorflow 2.0 , версии Tensorflow для процессора и графического процессора были упакованы вместе.

Чтобы заставить Tensorflow работать с вашим графическим процессором, вам необходимо загрузить cuDNN. В зависимости от того, какая у вас версия CUDA, вам нужно будет поместить некоторые заголовочные файлы и некоторые DLL-файлы в папку, в которой вы установили CUDA.