cupy.cuda.cudnn.CuDNNError: CUDNN_STATUS_INTERNAL_ERROR

Мои версии Cupy и Chainer следующие

Chainer: 4.1.0 NumPy: 1.15.4 CuPy: CuPy Версия: 4.1.0
Корень CUDA: /usr/local/cuda-9.0 Версия сборки CUDA: 9000 Версия драйвера CUDA: 9020 Версия CUDA Runtime: 9000
cuDNN Версия сборки: 7104 cuDNN Версия: 7104 NCCL Версия сборки: 2104

Я пытаюсь запустить тестовый скрипт по этой ссылке, и скрипт, который я использовал, выглядит следующим образом

python image_sheeping.py figure_skating/models/resnet_50_augmentation_no_noise_75_100/Resnet50SheepLocalizer_97305.npz \log -i figure_skating/valuation_dataset/test_images/22.png -g 0 -o valid__data / images / analysis

Я получаю сообщение об ошибке

Traceback (последний вызов был последним): файл "image_sheeping.py", строка 50, в bboxes, Scores = localizer.localize(processing_image)[:2] Файл "/home/rahul/Desktop/ Дипломная работа / код / ​​займы / займы /sheep/unsupervised_sheep_localizer.py", строка 43, локализовать bboxes, rois, результаты, visual_backprop = self.model.predict([обработанный_изображение], return_visual_backprop=return_visual_backprop) файл"/home/rahul/Desktop/Thesis/code/loan / loan /figure_skating/models/resnet_50_augmentation_no_noise_75_100/localizer.py", строка 102, в поле Предсказывать rois, bboxes = self(images) Файл" / home / rahul / Рабочий стол / Диссертация / код / ​​займы / займы / figure_skating / models / resnet_50_augmentation_no_noise_75_100 / py ", строка 144, в вызове h = self.feature_extractor(input_images, layer =['res5', 'pool5']) Файл"/home/rahul/.virtualenvs/loans/lib/python3.6/site-packages/chainer/links/model/vision/resnet.py", строка 198, в файле вызова h = func(h)"/home/rahul/.virtualenvs/loans/lib/python3.6/site-packages/chainer/links/connection/convolution_2d.py", строка 175, в группах вызовов =self.groups) Файл "/home/rahul/.virtualenvs/loans/lib/python3.6/site-packages/chainer/functions/connection/convolution_2d.py", строка 582, в свертке_2d y, = fnode.apply(args) Файл "/home/rahul/.virtualenvs/loans/lib/python3.6/site-packages/chainer/function_node.py", строка 258, в файле apply output = self.forward(in_data) "/home/rahul/.virtualenvs/loans/lib/python3.6/site-packages/chainer/function_node.py", строка 367, в прямом порядке возврат файла self.forward_gpu(входные данные) "/home/rahul/.virtualenvs/ loan / lib / python3.6 / site-packages / chainer / functions / connection / convolution_2d.py ", строка 161, в forward_gpu, возвращает файл self._forward_cudnn(x, W, b, y)" /home/rahul/.virtualenvs /loans/lib/python3.6/site-packages/chainer/functions/connection/convolution_2d.py ", строка 234, в _forward_cudnn auto_tune=auto_tune, tenor_core= тензор_core) Файл"cupy/cudnn.pyx", строка 598, в cupy.cudnn.convolution_forward Файл "cupy / cudnn.pyx", строка 33, в файле cupy.cudnn.get_handle "cupy / cuda / cudnn.p yx ", строка 473, в файле cupy.cuda.cudnn.create "cupy/cuda/cudnn.pyx", строка 446, в cupy.cuda.cudnn.check_status cupy.cuda.cudnn.CuDNNError: CUDNN_STATER_RUSDOR: 0%| | 0/1 [00:17 Traceback (последний вызов был последним): файл "/home/rahul/.virtualenvs/loans/lib/python3.6/site-packages/tqdm/_tqdm.py", строка 931, in del self Файл.close() "/home/rahul/.virtualenvs/loans/lib/python3.6/site-packages/tqdm/_tqdm.py", строка 1133, в закрытом файле self._decr_instances(self) "/ home / rahul /.virtualenvs/loans/lib/python3.6/site-packages/tqdm/_tqdm.py ", строка 496, в файле _decr_instances cls.monitor.exit()"/home/rahul/.virtualenvs/loans/lib/python3.6/site-packages/tqdm/_monitor.py", строка 52, в выходном файле self.join() Файл"/usr/lib/python3.6/threading.py", строка 1053, в повышении соединения RuntimeError(" не может присоединиться к текущей теме ") RuntimeError: невозможно присоединиться к текущей теме

Может кто-нибудь помочь мне решить ошибку?

1 ответ

Не могли бы вы перезапустить код со следующей переменной окружения?

export CUDNN_LOGDEST_DBG=cudnn_debug.log
export CUDNN_LOGINFO_DBG=1

Тогда, пожалуйста, поделитесь cudnn_debug.log файл (например, используя Gist).

Другие вопросы по тегам