exoにてCUDA_ERROR_UNSUPPORTED_PTX_VERSIONが発生する

Tinygrad で推論を実行しようとしたところ、以下のエラーが発生し動作しなかった。

RuntimeError: module load failed with status code 222: CUDA_ERROR_UNSUPPORTED_PTX_VERSION

ログには、

ptxas ... fatal : Unsupported .version 8.7; current version is '8.0'

とあり、コンパイルされた PTX バージョン (8.7) と、ptxas が対応しているバージョン (8.0) が合っていないことが原因だった。

解決策

NVIDIA ドライバをアップデートしたところ、不一致が解消され、エラーが消えた。実行したコマンドは以下。

sudo apt install cuda-drivers

アップデート後に OS を再起動した上で再度 Tinygrad を動かすと問題なく動作した。

まとめ

• 「CUDA_ERROR_UNSUPPORTED_PTX_VERSION」が起きた場合、古いドライバや CUDA Toolkit が原因で PTX バージョンが合わないケースが多い。
• ドライバを更新し、再起動してから再度実行すると解消される可能性が高い。