PyTorchをGPUサポート付きでインストールする方法


  1. CUDAドライバのインストール: 最初に、GPUを使用するために必要なCUDAドライバをインストールする必要があります。GPUのメーカーのウェブサイトから適切なCUDAドライバをダウンロードし、インストールします。

  2. cuDNNライブラリのインストール: PyTorchは、cuDNN(CUDA Deep Neural Network library)という高速化ライブラリを使用しています。cuDNNを利用するためには、NVIDIAのデベロッパーポータルからダウンロードしてインストールする必要があります。バージョンの互換性に注意してください。

  3. Anaconda (オプション): Anacondaを使用している場合は、以下のコマンドで仮想環境を作成します。

    conda create -n myenv python=3.9
    conda activate myenv
  4. PyTorchのインストール: GPUサポート付きのPyTorchをインストールするには、以下のコマンドを使用します。

    conda install pytorch torchvision torchaudio cudatoolkit=<バージョン>

    <バージョン>には、CUDAドライバと互換性のあるバージョンを指定します。例えば、CUDA 11.3を使用している場合は、cudatoolkit=11.3と指定します。

    もしAnacondaを使っていない場合は、pipコマンドを使用して以下のようにインストールします。

    pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu<バージョン>/torch_stable.html

    <バージョン>には、CUDAドライバのバージョンに対応する数字を指定します。

  5. 動作確認: インストールが完了したら、GPUが正しく使用されているかどうかを確認するために、以下のコードを実行してみてください。

    import torch
    print(torch.cuda.is_available())

    もしTrueが表示されれば、GPUが正しく検出されています。

以上が、PyTorchをGPUサポート付きでインストールする方法です。必要なライブラリやドライバを正しくインストールすることで、高速な深層学習の実行が可能になります。