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