RuntimeError: cublas runtime error : the GPU program failed to execute at C:/w/b/windows/pytorch/aten/src/THC/THCBlas.cu:272

解决问题思路

  1. torch版本与cuda版本不一致
  2. 电脑没有GPU可用或者电脑有多个cuda,但系统环境变量为设置相应cuda,导致系统找不到。

多半是torch版本或cuda问题 重新仔细安装!!

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐