看了无数篇博客都是牛头不对马嘴,忙活了半天才搞清楚
如果大家在用torch的过程中看见这样的错误。。。。淡定,不是你的问题,也不是代码问题。。。是破win10的原因
只需要在代码前面加上
torch.cuda.current_device()
torch.cuda._initialized = True
就可以正常训练了。。。心累啊
看了无数篇博客都是牛头不对马嘴,忙活了半天才搞清楚
如果大家在用torch的过程中看见这样的错误。。。。淡定,不是你的问题,也不是代码问题。。。是破win10的原因
只需要在代码前面加上
torch.cuda.current_device()
torch.cuda._initialized = True
就可以正常训练了。。。心累啊