RuntimeError: cuda runtime error (8) : invalid device function at pytorchtorchlibTHCgenericTHC

    技术2022-07-11  82

    一般是 CUDA 编译的问题,编译时候使用的参数与使用的GPU显卡不一致 在 Makefile 里编译的时候有关 CUDA 的参数

    arch=compute_61,code=sm_61

    参数里边的 61 指的是 GPU 显卡的计算能力,出现这种错误一般是计算能力与 GPU 显卡不符造成的,可以网上查询一下自己 GPU 显卡的计算能力,把这个数值改成自己 GPU 显卡的计算能力,重新编译就可以了

    查GPU计算能力:https://developer.nvidia.com/cuda-gpus

    Processed: 0.010, SQL: 9