I am getting desperate with Chainer because I'm not able to use it with GPU for about a week now. The error I am getting:
RuntimeError: CUDA environment is not correctly set up (see https://github.com/chainer/chainer#installation).No module named 'cupy.util'
Code to reproduce:
import chainer
chainer.cuda.to_gpu([0, 0])
Output of chainer.backends.cuda.available
is False
.
Working on Ubuntu 20.04 (I know, it is not the one from the recommended on Chainer's docs) inside WSL2. CUDA drivers 11.0. Output of 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 samples compile and work properly inside WSL2.
According to pip freeze
, cupy-cuda110
is installed within an (activated) virtual environment (but not detected, it seems). Chainer version 7.7.0 is installed.
Any ideas how to fix it?
Solution from https://github.com/chainer/chainer/issues/8582 did not seem to do the trick for me.