如何兼容cuda环境的训练软件，如ai-toolkit

你好，请教下，我用的沐曦官方的pytorch2.6的基础镜像，直接用原作者的aitoolkit源码启动，但是由于没有识别到英伟达显卡，导致无法继续，如何实现兼容cuda环境，我安装了cu-bridge没有效果，应该怎么解决，是否还要装英伟达的cuda-toolkit

尊敬的开发者您好，请提供以下信息
一、软硬件信息
1.服务器厂家:
2.沐曦GPU型号：
3.操作系统内核版本：
4.是否开启CPU虚拟化：
5.mx-smi回显：
6.docker info回显：
7.镜像版本：
8.启动容器命令：
9.容器内执行命令：
二、问题现象
请描述详细的问题现象日志。若日志过长，请上传附件(txt格式）。