UbuntuでPytorchをインストールする方法は色々な記事で紹介されていますが、躓くことが多かったので、自分なりにまとめて紹介することにしました。
STEP1 : Nvidiaドライバーをインストールせよ
nvidiaドライバーのインストールも様々な記事で紹介されているので問題ないと思います。
ubuntu-drivers devices
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-driver-510
sudo reboot
ubuntuにCUDA、nvidiaドライバをインストールするメモ - Qiita
使用しているGPUの確認 lspci | grep -i nvidia 現状入っているCUDA,nvidia-driverの確認 dpkg -l | grep nvidia dpkg -l | grep cuda ...
qiita.com
STEP2 : Pytorchをインストール
よくpytorchをインストールする前にシステム？の方にCUDAとcudnnをインストールする記事を見るのですが、今回紹介する方法はインストールしなくても良いです。
下のサイトを読むと、PyTorchにはCUDAが付属してくるようなので、システムの方のCUDAはあんまり関係ないみたいです。（ソースからビルドするときは別）
Difference between versions 9.2,10.1,10.2,11.0 of cuda for PyTorch 1.7
Recently,I need to install pytorch ,when I check out the website : install pytorch website It shows four different version 9.2,10.1,10.2,11.0 to choose ,And I h...
stackoverflow.com
PyTorch + CUDA 11.6
Hi, sir. I have one question: If I installed cuda 113 version pytorch, but my GPU is cuda 11.6. Is this mismatch of version affect the reproducibility of code. ...
discuss.pytorch.org
で、PyTorchのインストールですが、下のサイトから使いたいバージョンのPyTorchを選んでインストールします。
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
www.pytorch.org
私の場合は以下のようにしてインストールしてみました。
pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.htmlSTEP3 : GPUが使えるか確認せよ
最後にGPUが使えるところまで、できたかを確認します。
print(torch.cuda.is_available())
# TrueならOK
print(torch.version.cuda)
# 私の例なら11.1GPUが使えることを確認したら終了です！お疲れ様でした
コメント