使用torch查看cuda和cudnn版本

nvcc -V

import torch
print(torch.backends.cudnn.version())

我本地的是:

CUDA: 10.0
CUDNN:7605

posted @ 2022-02-11 15:55  乌蝇哥  阅读(738)  评论(0)    收藏  举报