pytorch报错CUDA error: invalid device function

Question

当我在服务器上跑github上的一个那个星最多的pytorch1版本的faster-rcnn时,服务器总是报错RuntimeError: CUDA error: invalid device function,但是我在其他服务器上跑的时候好像没有报这个错误,求教这是什么原因

Answer 1

检查显示卡是否兼容、驱动程序、cuda sdk和cudaa 的安装。