ubuntu22.04使用docker 镜像训练AI 不成功

2026-05-20 10:08:15 303 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

✨从零搭建 Ubuntu22.04 + Python3.11 + PyTorch2.5.1 GPU Docker 镜像并上传 Docker Hub 构建一个带 GPU 支持的深度学习开发镜像；使用 PyTorch 2.5.1（CUDA 12.4）环境；发布并托管 Docker 镜像至 Docker Hub；验证 GPU 环境正常运行！🎉 这将极大提升你的项目部署效率，尤其是在多人协作或跨平台部署场景中。阅读详情

根据官方指导手册，主机显卡驱动安装的是3.3.0版本，经过mthread-gmi测试，可以看到显卡的信息，应该是安装成功了的。然后在主机上安装MUSA SDK4.3.0等musa环境，根据官方的测试也是成功的。安装docker以及pull镜像也都没有问题。按照官方手册，使用docker镜像需要安装mt_container_toolkit，安装这个没有报错，应该也是成功的，mt_container与docker的绑定，按照官方手册给的命令，也是通过的，但是按照官方测试方法，启动docker镜像内部的mthreads-gmi时，出现红色警告说驱动或者container异常，mthreads-gmi运行没有输出，但也没有其他报错。test_musa.mu这个测试程序运行了，但输出看起来是随机的数。请问有人遇到这个问题吗?mt_container需要做其他操作吗，比如需要激活什么的。我是安装以后，直接使用命令与docker绑定一下，再也没有做其他操作了。

...全文