![](https://masaki-note.com/wp-content/uploads/2020/10/chris-ried-ieic5Tq8YMk-unsplash-1024x684.jpg)
【Pytorch】cudaのバージョンに合わせてPytorchをインストールする方法【GPU】
GPUが使える環境でPytorchをインストールしたいときの方法をまとめる。
cudaのバージョンによってインストールするバージョンが異なるので注意。
cudaに適したバージョンを探す
以下のページにcudaのバージョンに対応したPytorchのバージョンを調べることができる。
![](https://masaki-note.com/wp-content/uploads/cocoon-resources/blog-card-cache/1e9fe983d3406b3c9ac1a13d24004a42.jpg)
Previous PyTorch Versions
Installing previous versions of PyTorch
cudaのバージョンが10.1の場合は以下のように実行する。
pip3 install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
自分の環境ではPython3用にインストールする場合にはpip3を使うことになっている。
ここは環境によって適宜pipに変える
これでPytorchをインストールすることができた。
人気記事
人気記事はこちら。