GPU環境のセットアップ - Qiita業務でGPUサーバのセットアップを行なっていて色々詰まっているのでメモをまとめていこうと思う。 NVIDIA-DRIVERのインストールが終わってnvidai-dockerも動いたのでchainerを動かそうとしたらcuDNN関連で詰まってしまったのでその時のメモ。 環境情報 Ubuntu 16.04.4 LTS nvidia-driver 390.67 CUDA 9.1 Python3.5.2 つまったところ chainerでGPUを使って学習をしようとすると以下のエラーが出た。 CUDNN_STATUS_NOT_INITIALIZED 解決方法 cupyのインストールをミスっていたようである。 もともと以下の方法でcupyをインストールしていた。 pip install cupy これを改めて以下の方法でインストールするとエラーが出なくなった。 pip install cupy-cud
