cuda_error_invalid_ptx
原因有下面三个:
1. cuda version 和driver version不匹配,这个可能是最多的
2. cuda 不再支持此GPU。即GPU太老了,你又安装了太新的CUDA
3. .cu里写的CUDA code,有问题,比如buffer参数 (cuModuleLoadData参数)传递错误,share memory/threads超限。这个也会造成这个问题,但是网上提到的不多。
参考:
https://en.wikipedia.org/wiki/CUDA
https://stackoverflow.com/questions/60191445/ptx-jit-compilation-failed-from-cumoduleloaddata/67798656#67798656
https://discuss.tvm.apache.org/t/cuda-error-invalid-ptx-when-trying-to-run-single-conv2d-layer-after-compilation/712