こちらの記事が参考になります．
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…