对于cudnn可以在python中已有torch的前提下,输入
import torch
print(torch.backends.cudnn.version())
对于cuda toolkit,可以在命令提示符中输入
nvcc --version
对于cudnn可以在python中已有torch的前提下,输入
import torch
print(torch.backends.cudnn.version())
对于cuda toolkit,可以在命令提示符中输入
nvcc --version