現環境のCUDAとcuDNNバージョンを確認する方法
プログラムファイルにする必要もない程度ですが・・・
中身は
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
こんな感じで返してくれます。
もう何が何かわからん!というときにどうぞ
この記事が気に入ったらサポートをしてみませんか?