Pythonの環境や利用している仮想環境でインストールされているPyTorchのバージョンを知りたい場合があります。
この記事では、PyTorch (Torch) のインストールされているバージョンを確認する手順を紹介します。
以下のコマンドを実行します。
python -c "import torch; print(torch.__version__)"
実行すると以下の結果(一例)が表示されます。
2.1.2+cu121
2.6.0+cu124
前半部分がバージョン番号、+以降のcu...
が対応しているCUDAのバージョンとなります。
python を起動し、次のコマンドを実行します。
import torch
torch.__version__
コマンドを実行すると、pythonのバージョンが表示されます。