conda虚拟环境下查看cuda和cudnn版本的方法

conda虚拟环境下查看cuda和cudnn版本的方法?
1.如果没有安装pytorch,需要先安装

pip3 install pytorch

2.输入以下命令

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

注意：欢迎转载，转载时请注明来源