コンテンツへスキップ

CuPyエラー cuDNN is not enabled

cupyの利用に関して

Windowsでchainerを利用とした際に、cupy関係のエラーがでました。
自身のCuda利用環境に合わせたCupyを導入する必要があるようです。

C:\Users\miki\Anaconda3\envs\Own-Project\lib\site-packages\chainer\backends\cuda.py:98: UserWarning: cuDNN is not enabled.
Please reinstall CuPy after you install cudnn
(see https://docs-cupy.chainer.org/en/stable/install.html#install-cudnn).
  'cuDNN is not enabled.\n'

cupyインストール

pip install cupyすると、CUDA-8.0をターゲットとしてしまうみたいです。
なので、下記コマンドでCUDA-9.0用のcupyをインストールしてください。
pip install cupy-cuda90
cupyに関してはこちらに書いてありました。
CUDA 10.1を利用する場合
$ pip install cupy-cuda101
(Own-Project) C:\WINDOWS\system32>pip install cupy-cuda101
Collecting cupy-cuda101
Using cached https://files.pythonhosted.org/packages/14/80/35f5cb9858c3e914824b2a8c72db4c7a522e7aace0acfe469875dc05eedb/cupy_cuda101-6.2.0-cp36-cp36m-win_amd64.whl
Requirement already satisfied: six>=1.9.0 in c:\users\miki\anaconda3\envs\own-project\lib\site-packages (from cupy-cuda101) (1.12.0)
Requirement already satisfied: fastrlock>=0.3 in c:\users\miki\anaconda3\envs\own-project\lib\site-packages (from cupy-cuda101) (0.4)
Requirement already satisfied: numpy>=1.9.0 in c:\users\miki\anaconda3\envs\own-project\lib\site-packages (from cupy-cuda101) (1.16.4)
Installing collected packages: cupy-cuda101
Successfully installed cupy-cuda101-6.2.0

利用環境に合わせたCupyを導入し、エラーがなくなりました。

Cupy、Chainerの動作環境

chainer.print_runtime_info()
print('GPU:', chainer.cuda.available)
print('cuDNN:', chainer.cuda.cudnn_enabled)
Platform: Windows-10-10.0.17134-SP0
Chainer: 5.3.0
NumPy: 1.16.4
CuPy:
  CuPy Version          : 6.2.0
  CUDA Root             : C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
  CUDA Build Version    : 10010
  CUDA Driver Version   : 10010
  CUDA Runtime Version  : 10010
  cuDNN Build Version   : 7500
  cuDNN Version         : 7500
  NCCL Build Version    : None
  NCCL Runtime Version  : None
iDeep: Not Available
GPU: True
cuDNN: True

コメントを残す

メールアドレスが公開されることはありません。 が付いている欄は必須項目です

CAPTCHA