pytorch使用cuda报错RuntimeError: CUDA error: unknown error
pytorch使用cuda报错RuntimeError: CUDA error: unknown error加上torch.cuda.current_device()例如import osimport torchimport torch.nn as nnimport argparsefrom sklearn import metricsfrom torch.utils....
原创
2019-12-17 19:40:13 ·
4279 阅读 ·
1 评论