在github上下了一個chainer的程序,安裝chainer和cupy-cuda91後立即運行程序出現下面異常:
catastrophic error: cannot open source file "cuda_fp16.h"
在網上搜解決辦法,看了chainer官網的安裝教程,也看了一些博客(chainer相關的東西很少,很多也是重複的)都沒能找到解決問題;同時網上的一些簡單測試程序測試cupy和chainer都顯示也安裝成功,但運行我的程序就是出錯。
後面自己摸索程序出錯提示找到解決辦法,下面介紹這個對我有效的辦法:
1、修改cupy安裝目錄下的一個文件,叫compiler.py;此文件在我電腦上的安裝目錄是
/home/alloymei/anaconda3/envs/tensorflow/lib/python3.5/site-packages/cupy/cuda/compiler.py
2、在此程序上293行添加一行代碼:backend='nvcc'