このサイト(http://qiita.com/JeJeNeNo/items/05e148a325192004e2cd)に従ってCUDAとcuDNNをインストールしました。(CUDAは自分の環境に合わせて、cuda-repo-ubuntu1604_8.0.*_amd64 ではなく、cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb をダウンロードしました)
しかし、実際にCUDAを使ったプログラムを実行すると、以下のようなエラーが出力されます。
/usr/local/lib/python2.7/dist-packages/chainer/cuda.py:92: UserWarning: cuDNN is not enabled. Please reinstall chainer after you install cudnn (see https://github.com/pfnet/chainer#installation). 'cuDNN is not enabled.\n' (中略) cupy.cuda.runtime.CUDARuntimeError: cudaErrorMemoryAllocation: out of memory
cuDNNはすでにインストールされていますし、chainerもまた別のこちらのサイト(http://robotics4society.com/2016/06/17/ubuntu-chainer/)を参考にインストールし、GPUも認識されているようなので、何がいけないのか困っている状況です。
インストール後に何かしらの方法でenableにする必要があるのでしょうか?
詳しい方がいらっしゃいましたら、ご教授よろしくお願いいたします。
*ubuntu 16.04 です。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。