前提: 安装了pytorch
打开一个终端,依次输入以下命令
python3
import torch
print(torch.version.cuda)
print(torch.backends.cudnn.version())
前提: 安装了pytorch
打开一个终端,依次输入以下命令
python3
import torch
print(torch.version.cuda)
print(torch.backends.cudnn.version())