看了无数篇博客都是牛头不对马嘴,忙活了半天才搞清楚
如果大家在用torch的过程中看见这样的错误。。。。淡定,不是你的问题,也不是代码问题。。。是破win10的原因

只需要在代码前面加上
torch.cuda.current_device()
torch.cuda._initialized = True
就可以正常训练了。。。心累啊
本文解决了在Windows 10系统中使用PyTorch进行深度学习项目时遇到的CUDA初始化错误问题。只需在代码开头加入两行特定的PyTorch CUDA API调用,即可避免此错误,使模型训练恢复正常。
看了无数篇博客都是牛头不对马嘴,忙活了半天才搞清楚
如果大家在用torch的过程中看见这样的错误。。。。淡定,不是你的问题,也不是代码问题。。。是破win10的原因

只需要在代码前面加上
torch.cuda.current_device()
torch.cuda._initialized = True
就可以正常训练了。。。心累啊
2877
1337
4483
2101