Ubuntu PyTorch インストール手順と使い方【簡単解説】

1年 ago
桜, 春樹
1 minute

UbuntuにPyTorchをインストールするには、condaまたはpipを使用することができます。PyTorchをcondaでインストールする手順は次の通りです：

  1. 最初に、condaがインストールされていることを確認してください。condaがインストールされていない場合、以下の手順に従ってインストールできます。
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
source ~/.bashrc
  1. 新しいconda環境を作成し、その環境をアクティブにします。
conda create -n myenv
conda activate myenv
  1. PyTorchをcondaを使用してインストールする。
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

Pythonスクリプト内でPyTorchを使用する準備が整いました。以下は簡単な例です：

import torch

x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([4.0, 5.0, 6.0])

z = x + y

print(z)

上記のコードを実行すると、2つのテンソルの合計が出力されます。これはPyTorchが正常にインストールされていて使用されていることを示しています。

PyTorchをインストールする際は、pipを使用することもできます。PyTorchをpipでインストールする手順は以下の通りです：

  1. 最初に、pipがインストールされていることを確認してください。pipがインストールされていない場合は、以下のコマンドを使用してインストールできます：
sudo apt update
sudo apt install python3-pip
  1. PyTorchをインストールするためにpipを使用する:
pip install torch torchvision torchaudio

インストールが完了したら、上記の例を参考にしてPyTorchを使用することができます。

これがあなたがUbuntuでPyTorchをインストールして使用するのに役立つといいです。

bannerAds