Как проверить, является ли версия tenorflow 2.1.0 gpu или cpu? - PullRequest
0 голосов
/ 28 февраля 2020

Я использую: pyhton 3.6 тензор потока 2.1.0 Я хотел бы запустить свой код python в режиме графического процессора, но я думаю, что у меня есть проблема в CUDA, потому что я нахожу это, когда я запускаю свой код:

2020-02-28 14:07:28.180007: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer_plugin.so.6'; dlerror: libnvrtc.so.10.1: cannot open shared object file: No such file or directory
2020-02-28 14:07:28.180021: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:30] Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
/home/hp/anaconda3/lib/python3.6/site-packages/h5py/__init__.py:34: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters
2020-02-28 14:07:28.942588: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libcudart.so.10.1'; dlerror: libcudart.so.10.1: cannot open shared object file: No such file or directory
2020-02-28 14:07:28.942612: 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.
WARNING:tensorflow:From hdf.py:19: is_gpu_available (from tensorflow.python.framework.test_util) is deprecated and will be removed in a future version.
Instructions for updating:
Use `tf.config.list_physical_devices('GPU')` instead.
2020-02-28 14:07:29.145709: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
2020-02-28 14:07:29.151283: I tensorflow/core/platform/profile_utils/cpu_utils.cc:94] CPU Frequency: 4007925000 Hz
2020-02-28 14:07:29.151569: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x556386ef65e0 initialized for platform Host (this does not guarantee that XLA will be used). Devices:
2020-02-28 14:07:29.151587: I tensorflow/compiler/xla/service/service.cc:176]   StreamExecutor device (0): Host, Default Version
2020-02-28 14:07:29.153249: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libcuda.so.1'; dlerror: libcuda.so.1: cannot open shared object file: No such file or directory
2020-02-28 14:07:29.153265: E tensorflow/stream_executor/cuda/cuda_driver.cc:351] failed call to cuInit: UNKNOWN ERROR (303)
2020-02-28 14:07:29.153284: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:169] retrieving CUDA diagnostic information for host: hp-870-102nk
2020-02-28 14:07:29.153294: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:176] hostname: hp-870-102nk
2020-02-28 14:07:29.153330: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:200] libcuda reported version is: Not found: was unable to find libcuda.so DSO loaded into this program
2020-02-28 14:07:29.153360: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:204] kernel reported version is: 418.56.0

...