记录下linux系统conda虚拟环境下查看cuda和cudnn版本的方法

前提: 安装了pytorch
打开一个终端,依次输入以下命令

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

你可能感兴趣的:(Linux系统,c++,计算机视觉)