nvidia-smi
とnvcc -V
が使える環境下において
何も考えず脳死でPytorchをpip install torch
と実行したときに,libcudart.soが無いと怒られます。
これはNVIDIA GPU周りのトラブルではなく、脳死でインストールした自分のせいです。
対処法
-
nvcc -V
でcudaのバージョンを確認してください。 -
PyTorchのページに行き,cudaのバージョンと合うものを選んでインストールしてください。
例:cuda:11.8の場合
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118
-
python
コマンドを入力し,対話モードにします。 -
import torch
をした後、torch.cuda.is_availabe()
を入力し、True
と出ればPytorchが使えます。