标签：operations ignored gu errors keras py __ File line

问题现象

2021-06-04 16:54:01.790711: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
  0%|                                                                                    | 0/102 [00:02<?, ?it/s]
Traceback (most recent call last):
  File "get_dr_txt.py", line 120, in <module>
    frcnn.detect_image(image_id,image)
  File "get_dr_txt.py", line 64, in detect_image
    rpn_pred = self.model_rpn.predict(photo)
  File "/home/user/anaconda3/envs/gu_keras/lib/python3.7/site-packages/keras/engine/training.py", line 1462, in predict
    callbacks=callbacks)
  File "/home/user/anaconda3/envs/gu_keras/lib/python3.7/site-packages/keras/engine/training_arrays.py", line 324, in predict_loop
    batch_outs = f(ins_batch)
  File "/home/user/anaconda3/envs/gu_keras/lib/python3.7/site-packages/tensorflow_core/python/keras/backend.py", line 3476, in __call__
    run_metadata=self.run_metadata)
  File "/home/user/anaconda3/envs/gu_keras/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1472, in __call__
    run_metadata_ptr)
tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
  (0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
         [[classification/Reshape/_1971]]
    frcnn.detect_image(image_id,image)  File "get_dr_txt.py", line 64, in detect_image    rpn_pred = self.model_rpn.predict(photo)  File "/home/user/anaconda3/envs/gu_keras/lib/python3.7/site-packages/keras/engine/training.py", line 1462, in predict    callbacks=callbacks)
  File "/home/user/anaconda3/envs/gu_keras/lib/python3.7/site-packages/keras/engine/training_arrays.py", line 324, in predict_loop
    batch_outs = f(ins_batch)
  File "/home/user/anaconda3/envs/gu_keras/lib/python3.7/site-packages/tensorflow_core/python/keras/backend.py",line 3476, in __call__    run_metadata=self.run_metadata)  File "/home/user/anaconda3/envs/gu_keras/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1472, in __call__
    run_metadata_ptr)
tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
  (0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
         [[{{node conv1/convolution}}]]
         [[classification/Reshape/_1971]]
  (1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
         [[{{node conv1/convolution}}]]
0 successful operations.
0 derived errors ignored.

解决方式
设置了一下GPU的使用情况

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"  #实现卡号匹配
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

标签：operations,ignored,gu,errors,keras,py,__,File,line
来源： https://blog.csdn.net/qq_43120531/article/details/117567698

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。