UbuntuでPyTorchをインストールして使用する方法は何ですか。

1年 ago
結衣, 春花
1 minute

UbuntuにPyTorchをインストールして使用するには、次の手順に従うことができます。

  1. Anacondaのインストール: ターミナルで以下のコマンドを実行して、Anacondaをダウンロードしてインストールします。これはPython環境を管理するためのツールです。
  2. wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.shを実行して、chmod +x Anaconda3-2021.05-Linux-x86_64.shで権限を変更し、./Anaconda3-2021.05-Linux-x86_64.shを実行してください。
  3. インストール中は指示に従って設定を行ってください。インストールが完了したら、source ~/.bashrcコマンドを実行してAnacondaを有効にしてください。
  4. Anaconda環境を新たに作成します：ターミナルで以下のコマンドを実行し、pytorchという名前の新しい環境を作成し、その環境をアクティブにします。
  5. conda create -n pytorch python=3.8を実行してください
    conda activate pytorch を実行してください
  6. PyTorchのインストール：PyTorchの環境をアクティブにした状態で、以下のコマンドを実行してPyTorchをインストールしてください。
  7. pytorch torchvision torchaudio cudatoolkit=11.1 をconda-forgeとpytorchのチャンネルからインストールします。
  8. Pythonの対話型環境で以下のコマンドを実行して、PyTorchのインストールが成功しているか確認してください。
  9. import torch
    バージョンを出力する（print(torch.__version__)）
  10. PyTorchのバージョン番号が正常に表示されたら、インストールが成功していることを示します。

現在、PyTorchを使った深層学習のタスクができます。PyTorchの公式ドキュメントを参照して、PyTorchの機能やAPIの使い方を学ぶことができます。

bannerAds