我在使用曙光DCU时，连接节点后(yolo) [ac2ax5rex7@e10r4n02 yolov7-main]$ python Python 3.7.12 | packaged by conda-forge | (default, Oct 26 2021, 06:08:21) [GCC 9.4.0] on linux Type "help", "copyright", "credits" or "license" for more information. >>> import torch >>> torch.cuda.is_available() Killed

这个问题可能是由于内存不足导致的。当您在使用曙光DCU时，连接到节点并尝试运行 yolo 模型时，可能会出现内存不足的情况，导致 Python 进程被 kill 掉。

您可以尝试使用更小的 batch size 或减少模型的输入尺寸来减少内存消耗。或者，您也可以尝试使用更高内存的节点来解决这个问题。如果问题仍然存在，您可以尝试使用其他的深度学习框架，例如 TensorFlow 或 MXNet，来提高运行效率和减少内存消耗。