Skip to content

README.md

Latest commit

 

History

History
37 lines (28 loc) · 1.21 KB

README.md

File metadata and controls

37 lines (28 loc) · 1.21 KB

查看驱动版本

# 查看到驱动版本具体日期
sudo clinfo|grep Driver 

# 可查看驱动大版本号
mthreads-gmi

驱动版本和镜像版本对应关系

Docker Image List
Driver Version GPU Docker Image
Driver Version 20241025 S4000 registry.mthreads.com/mcconline/musa-pytorch-release-public:rc3.1.0-v1.3.0-S4000-py38
Driver Version 20241025 S3000 registry.mthreads.com/mcconline/musa-pytorch-release-public:rc3.1.0-v1.3.0-S3000-py38
Driver Version 20241025 S80 registry.mthreads.com/mcconline/musa-pytorch-release-public:rc3.1.0-v1.3.0-S80-py38

NOTE: Python3.10 版本镜像请使用 py310 替换上述镜像 tag 中 py38.

请按照实际驱动环境，参考上述表格，更换启动docker命令中镜像名

# S4000 Python3.8
docker run -it --privileged --pull always --network=host --name=torch_musa_test -v /data:/data --env MTHREADS_VISIBLE_DEVICES=all --shm-size=80g registry.mthreads.com/mcconline/musa-pytorch-release-public:rc3.1.0-v1.3.0-S4000-py38 /bin/bash

容器环境检查

# 在容器内执行
cd tutorial_on_musa/setup_musa/check
bash test_musa.sh