-
CUDAドライバのインストール: 最初に、GPUを使用するために必要なCUDAドライバをインストールする必要があります。GPUのメーカーのウェブサイトから適切なCUDAドライバをダウンロードし、インストールします。
-
cuDNNライブラリのインストール: PyTorchは、cuDNN(CUDA Deep Neural Network library)という高速化ライブラリを使用しています。cuDNNを利用するためには、NVIDIAのデベロッパーポータルからダウンロードしてインストールする必要があります。バージョンの互換性に注意してください。
-
Anaconda (オプション): Anacondaを使用している場合は、以下のコマンドで仮想環境を作成します。
conda create -n myenv python=3.9 conda activate myenv
-
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ドライバのバージョンに対応する数字を指定します。
-
動作確認: インストールが完了したら、GPUが正しく使用されているかどうかを確認するために、以下のコードを実行してみてください。
import torch print(torch.cuda.is_available())
もし
True
が表示されれば、GPUが正しく検出されています。
以上が、PyTorchをGPUサポート付きでインストールする方法です。必要なライブラリやドライバを正しくインストールすることで、高速な深層学習の実行が可能になります。