Pytorch错误集锦
RuntimeError: Expected one of cpu, cuda, mkldnn, opengl, opencl, ideep, hip, msnpu device type at start of device string: 0在python\lib\site-packages\torch\cuda_init_.py文件中的set_device方法修改device = torch.device("cuda:0" if torch.cuda.is_available() else ".
原创
2020-10-13 21:53:23 ·
6794 阅读 ·
1 评论