使用CUDA训练时会导致Kernel崩溃
收藏
进行简单的Tensor运算时,使用cuda可以正常运行。所以应该不是cuda安装的问题。
但是在使用数据集训练一个CNN时,使用cpu可以正常运行,使用cuda时,则总会导致kernel崩溃,提示:
Canceled future for execute_request message before replies were done The Kernel crashed while executing code in the the current cell or a previous cell. Please review the code in the cell(s) to identify a possible cause of the failure. Click here for more info. View Jupyter log for further details.
0
收藏
请登录后评论