元々Windows 10がインストールされていて、それを潰してUbuntuを入れていたのだが、CUDAのバージョンがどうにもならないのでWindowsを再インストールすることとした。一度インストールし、VisualStudioその他のツールもインストールしたところで、久々にBIOSをアップデートしようとアップデートしたところ、何故かSSDからブートできなくなった。おそらくBIOSのLegacy/UEFIの問題かと思われるが、設定を変えても結局起動できず、やむなくWindowsを再々インストール。
その後、Python、CUDA 11.3等もインストールしてPyTorchのインストールへ。pip3を用いて公式サイト通りにインストールしたところ、無事インストールが完了し、スクリプトも一通り走った。
おそらくCUDAのバージョン使い分けもWindowsの方が容易と思われる、当分はこのままで行こうと思う。