操作系统ubuntu
显卡AMD 6750
已经配置了rocm
torch.cuda.is_available()运行结果为True
但是在运行 print(torch.rand(3,3).cuda()) 时显示 进程已结束,退出代码为 139 (interrupted by signal 11: SIGSEGV)
引用 皆我百晓生 小程序回复内容作答:
这个错误代码通常表示遇到了段错误(segmentation fault)。这可能是由于不兼容的驱动程序或库版本或其他的硬件问题引起的。在使用深度学习框架时,通常建议使用NVIDIA的GPU,因为AMD的GPU在深度学习方面的支持相对较差。
要解决这个问题,可以尝试以下几个步骤:
此外,确保操作系统和深度学习框架的最新版本可用,以确保使用最新的修复和功能。如果问题仍然存在,建议更详细地查看错误日志以获取更多的信息,或者在相关的开发者社区中咨询,以获得更具体的帮助。