[ツール] PyTorchのインストール(CUDAのバージョンに合わせて)

CUDA Toolkit アーカイブ (PyTorch のすべてのバージョンを含む)

必要な CUDA バージョンと PyTorch バージョンに応じて、対応するダウンロード コマンドを見つけてください。ここでは、PyTorch1.10.1 CUDA 11.3 を例として取り上げます。

Linux と Windows

# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

# CPU Only
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cpuonly -c pytorch