pytorch GPU対応方法

こんばんは。

今日、間違ってPytorchの仮想環境を壊してしまい、焦ってしまいました。

pytorchがGPUを認識しない

これは焦りました。最初、GPU対応版ではなく適当にcondaでパッケージをインストールしたんですが、GPUをプログラムが認識しないって症状がおきました。

このため、pytorchの公式サイトを確認し再インストールしました。

pytorch公式サイト

インストール方法

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

pytorchが正常動作動作しているのかを確認する方法

$python

import torch
x = torch.rand(5, 3)
print(x)

出力結果が下記がでると正常です。

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

代替方法は以下の通りです。

import torch
torch.cuda.is_available()

私はUbuntuを利用してますが、バージョンの管理方法は以下の通りでした。

cat /etc/os-release

NAME="Ubuntu"
VERSION="20.04 LTS (Focal Fossa)"
ID=ubuntu
ID_LIKE=debian
PRETTY_NAME="Ubuntu 20.04 LTS"
VERSION_ID="20.04"
HOME_URL="https://www.ubuntu.com/"
SUPPORT_URL="https://help.ubuntu.com/"
BUG_REPORT_URL="https://bugs.launchpad.net/ubuntu/"
PRIVACY_POLICY_URL="https://www.ubuntu.com/legal/terms-and-policies/privacy-policy"
VERSION_CODENAME=focal
UBUNTU_CODENAME=focal