現環境のCUDAとcuDNNバージョンを確認する方法

プログラムファイルにする必要もない程度ですが・・・

cuda-kakunin.py

196 Bytes

ファイルダウンロードについて

ダウンロード

中身は

import torch
print('GPU の利用状況 :', torch.cuda.is_available())
print('CUDA のバージョン :', torch.version.cuda)
print('cuDNN のバージョン :', torch.backends.cudnn.version())

これだけです。
使い方は
python cuda-kakunin.py

GPU の利用状況 : True
CUDA のバージョン : 12.4
cuDNN のバージョン : 90100

こんな感じで返してくれます。
もう何が何かわからん！というときにどうぞ