Pytorch常用代码段集锦
本实验内容基于Pytorch 1.51 导入包和版本查询# -*- coding:utf-8 -*-import torchimport torch.nn as nnimport torchvisionif __name__=="__main__": print(torch.__version__) # torch 版本号 print(torch.version.cuda)# torch cuda 版本号 print(torch.backends.cudnn.version(
