(A2)查看pytorch、cuda、cudnn版本

查看pytorch、cuda、cudnn版本

参考资料

pytorch

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

linux

nvcc --version
nvcc -V

你可能感兴趣的:(常用指令积累,pytorch,深度学习,python)