PyTorchのバージョンする方法をいくつか紹介する。
__version__属性を使う方法
PyTrochのversion属性を使うことで、バージョン番号を取得する。
import torch
print(torch.__version__)
# 1.12.1+cu113PyTorchのバージョンとCUDAバージョンが出力として表示されます。上記の場合（1.12.1+cu113）、PyTorchのバージョンは1.12.1で、CUDAのバージョンは11.3であることを表しています。
torchaudio、torchvision、torchtextなども同様の方法で確認することができます。
import torchaudio
import torchvision
import torchtext
print(torchaudio.__version__)
print(torchvision.__version__)
print(torchtext.__version__)
# 0.12.1+cu113
# 0.13.1+cu113
# 0.13.1pipを使う方法
pip経由でインストールしている場合、pipコマンドで確認することができます。
pip list
pip listを使うことで、バージョンを表示させることができます。pip listだけを使うと全てのライブラリのバージョンが表示されるため、grepを使ってtorchで表示するライブラリを指定します。
pip list | grep torch
# torch 1.12.1+cu113
# torchaudio 0.12.1+cu113
# torchsummary 1.5.1
# torchtext 0.13.1
# torchvision 0.13.1+cu113pip show
pip showを使うとバージョン情報以外にもパッケージの詳細情報が表示されます。
pip show torch
# Name: torch
# Version: 1.12.1+cu113
# Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
# Home-page: https://pytorch.org/
# Author: PyTorch Team
# Author-email: packages@pytorch.org
# License: BSD-3
# Location: /usr/local/lib/python3.7/dist-packages
# Requires: typing-extensions
# Required-by: torchvision, torchtext, torchaudio, fastai関連記事、参考資料
PyTorchの入門書で、GPUの利用方法、ネットワークの構築方法や転移学習まで幅広く書かれていてオススメです。
コメント