ご自身のパソコンにPytorchをインストールの後の確認、AWSで機械学習用のサーバのインスタンスを立てたとき、Google Colaboratoryなど既設のサーバを使う時など、インストールされているPytorchのバージョンを確認したいことがあります。
そんな時に簡単にバージョンを確認する方法をご紹介します。
Pytorchのバージョンを確認する方法
パソコンに入っているPytorchのバージョンを確認する方法
jupyter notebook の起動
Anacondaまたは、ご自身でインストールしたJupyter nootbookを起動してください。
バージョンの確認
以下の通り、torchをインポートしてバージョンを確認するコマンドを実行して下さい。
import torch torch.__version__
上記のコマンドを実行すると例えば ’1.8.0’ と表示されます。
これがPytorchのバージョンを表しています。
実際の値はお使いの環境にインストールされているものが表示されます。
尚、もし例えば ’1.8.0+cu111’ と表示された場合
おめでとうございます。
これはあなたがお使いの環境ではNVIDIAの提供するGPUが使える環境となっていることを示しており、’cu111’ の部分はCUDAのバージョンを示しています。
尚、この表記には小数点が省略されており、この場合バージョン ’111’ ではなく ’11.1’ であることを表しています。
Google ColaboratoryでのPytorchのバージョンを確認する方法
「パソコンに入っているPytorchのバージョンを確認する方法」でご紹介した方法で確認できます。
Google ColaboratoryではGPUが使える環境となっているので、必ずGPUのバージョンも表示されます。
AWSのインスタンスでのPytorchのバージョンを確認する方法
基本的にこれまでと同じですが、もしjupyter notebookがインストールされていない環境をお使いの場合は、コンソールで ‘python’ を起動し、これまでと同じようにコマンドを入力し実行すると確認できます。
実行例は下記の通り。
赤字の部分が入力して実効する部分です。
% python Python 3.6.12 |Anaconda, Inc.| (default, Sep 8 2020, 17:50:39) [GCC Clang 10.0.0 ] on darwin Type "help", "copyright", "credits" or "license" for more information. >>> import torch >>> torch.__version__ '1.7.0' >>>
以上、Pytorchのバージョンの確認方法でした。
