PyTorchのインストール

PC環境

• OS: Ubuntu 20.04 LTS

• GPU: NVIDIA GeForce RTX2070 SUPER

• NVIDIAドライバ: 535.54.03

• CUDAバージョン: 12.2

• cuDNNバージョン: 8.9.2.26

• Python: 3.8.10

• PyTorch: 1.12.1

GPUセットアップはこちらの記事を参照：

仮想環境セットアップ

$ python3 --version
Python 3.8.10

$ sudo apt update
$ sudo apt install python3.8-venv
$ python3 -m venv .venv_pytorch-1.12
$ source .venv_pytorch-1.12/bin/activate

$ pip install -U pip
$ pip --version
pip 23.1.2 from /home/XXX/.venv_tensorflow-2.11/lib/python3.8/site-packages/pip (python 3.8)

PyTorchをインストール

$ pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

インストール確認

$ python
>>> import torch
>>> print(torch.__version__, torch.cuda.is_available())
1.12.1+cu116 True
>>> x = torch.rand(5, 3)
>>> print(x)
tensor([[0.0021, 0.2511, 0.7921],
        [0.3906, 0.7013, 0.9575],
        [0.0948, 0.1497, 0.9910],
        [0.8271, 0.4473, 0.7397],
        [0.5587, 0.6977, 0.1337]])

1つ目のprint文でバージョンとTrueが返ってくれば、GPUもOK.
2つ目のprint文でtensorが返ってくればOK.

---

参考URL:

Previous PyTorch Versions