Tensorflow-CPU:加载模型时出错

时间:2020-03-27 16:53:58

标签: tensorflow

从.h5文件加载模型时遇到以下错误。

  • tf版本:2.1.0
  • CUDA版本:V9.1.85
  • python版本:3.7.3

尝试了“ os.environ ['CUDA_VISIBLE_DEVICES'] ='0'”,但没有成功

错误:

2020-03-27 22:11:32.164829: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library libcuda.so.1
2020-03-27 22:11:32.165699: E tensorflow/stream_executor/cuda/cuda_driver.cc:313] failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected
2020-03-27 22:11:32.165717: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:156] kernel driver does not appear to be running on this host (Lenovo-ideapad-310-15IKB): /proc/driver/nvidia/version does not exist
2020-03-27 22:11:32.165938: I tensorflow/core/platform/cpu_feature_guard.cc:143] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
2020-03-27 22:11:32.192087: I tensorflow/core/platform/profile_utils/cpu_utils.cc:102] CPU Frequency: 2899885000 Hz
2020-03-27 22:11:32.192311: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x7f7364000b20 initialized for platform Host (this does not guarantee that XLA will be used). Devices:
2020-03-27 22:11:32.192332: I tensorflow/compiler/xla/service/service.cc:176]   StreamExecutor device (0): Host, Default Version
Xception load success

0 个答案:

没有答案