Я впадаю в отчаяние с 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 сделало не похоже, чтобы сделать трюк для меня.