運行chainer程序出現catastrophic error: cannot open source file "cuda_fp16.h"

      在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'

                                   

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章