Chainer: нет модуля с именем «cupy.util»

Я впадаю в отчаяние с Chainer, потому что я не могу использовать его с графическим процессором уже около недели. Я получаю ошибку: RuntimeError: CUDA environment is not correctly set up (see https://github.com/chainer/chainer#installation).No module named 'cupy.util'

Код для воспроизведения:

import chainer
chainer.cuda.to_gpu([0, 0])

Выход chainer.backends.cuda.available равен False.

Работаю над Ubuntu 20.04 (я знаю, что это не та версия, которая рекомендована в документации Chainer) внутри WSL2. Драйверы CUDA 11.0. Выход nvcc -V:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Wed_Jul_22_19:09:09_PDT_2020
Cuda compilation tools, release 11.0, V11.0.221
Build cuda_11.0_bu.TC445_37.28845127_0

Образцы CUDA правильно компилируются и работают внутри WSL2.

Согласно pip freeze, cupy-cuda110 установлен в (активированной) виртуальной среде (но, похоже, не обнаружен). Установлен Chainer версии 7.7.0.

Есть идеи, как это исправить?

Решение с сайта https://github.com/chainer/chainer/issues/8582 сделало не похоже, чтобы сделать трюк для меня.


person Valeria    schedule 01.03.2021    source источник
comment
Пожалуйста, используйте CuPy v7.8.0 для серии Chainer v7.   -  person kmaehashi    schedule 02.03.2021


Ответы (1)


Сообщение об ошибке очень ясное. Просто измените L69 из backends/cuda.py:

from cupy.util import PerformanceWarning as _PerformanceWarning

to

from cupy._util import PerformanceWarning as _PerformanceWarning

вместе с решением из #8582 все будет работать нормально.

person H3NT41    schedule 06.06.2021