任意のCUDAバージョンのPyTorchをPoetryでインストールする

2023/07/03追記 Poetry 1.5.1より、指定したソースからのインストールが容易になったようです。下記記事を参考ください。 Poetry1.5.1からGPU版のPytorchのインストールが簡単になりました PyTorchからGPUを利用するためには、CUDAのバージョンに合ったパッケージをインストールする必要があります。PyTorchの"Get Started"ページでは、プラットフォームやバージョン、パッケージマネージャーを選択すると、それに応じたインストールコマンドが表示される仕組みがありますが、ここにはPoetryは含まれていません。 そこでこの記事では、Poetryを利用しているプロジェクトにおいて、任意のCUDAバージョンのPyTorchパッケージをインストールする方法を解説します。 方法 実現する方法はいくつか存在するので、それぞれ解説していきます。ここでは、例