pytorch1.9.0におけるcpp_extensionのデフォルトのCUDAバージョンが11.0である問題について

今日、CUDAExtension を使用してオペレーターをコンパイルしたところ、デフォルトの CUDA パスがバージョン 11.0 を指していることがわかりました。私の CUDA バージョンは 11.1 でした。

これにより、nvcc でコンパイルするときに「No such file or directory」エラーが報告されます。

現時点では、次のように CUDA のパスをインストールのパスに指定するだけで済みます。

torch.utils.cpp_extension.CUDA_HOME = "/usr/local/cuda-11.1"

CUDA パスを修正できます。