请问用pytorch跑cuda的程序总是中断,Process finished with exit code -1073741819 (0xC0000005)出现这个问题,调试的时候是loss backward()这里出问题了,但是测试cuda一切正常,求解决
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name())
print(torch.cuda.current_device())
#True
#1
#GeForce RTX 2060 SUPER
#0
是cuda版本不匹配的原因。我电脑是10以上的cuda,之前装的是9.2的,后面装了10.1的就可以运行了
是的,已解决,是版本不匹配的原因