运行A程序时，出现如下错误。
考虑原因：A程序在语料库ATIS中运行过，没有发生问题，但是之后运行了另一个程序B时，修改过gpu，如下所示：
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
如今在语料库Snips上运行A程序，却发生以下错误：
因此认为是GPU没有选择好，在A程序的main函数中添加一行程序后，顺利解决：
torch.cuda.current_device()
RuntimeError:CUDA error:unknown error
最新推荐文章于 2024-08-01 09:52:55 发布
在运行A程序时，因GPU选择不当导致错误。此前在语料库ATIS中运行正常，但在Snips上出现问题。通过在A程序main函数中添加torch.cuda.current_device()调用，成功解决了GPU兼容性问题。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
660
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
