PyTorchでGPUを使うために
2024-08-02
こちらの記事が参考になります．
https://qiita.com/rikinumata/items/b3729ad5144a3e98c6b4
1.PyTorchが要求するCUDAのバージョンの確認
Pythonでありがちな事象ですが，バージョンの組み合わせを熟慮してから取り掛かる必要があります．
PyTorchのウェブサイト(https://pytorch.org/)で以下のようにインタラクティブに実行すべきpipコマンドを教えてくれますが，これだけ見てpipを実行するとハマります．
つまり，ここではバージョンの確認だけします．
2.自分の環境で使えるCUDAのバージョン確認
NVIDIAのウェブサイト https://developer.nvidia.com/cuda-gpus…