a946079208
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix nccl bug in gpu docker container

This commit is contained in:
wuweikang 2020-11-07 16:00:12 +08:00
parent 3e77f19f31
commit 44c4eaa12c
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -172,8 +172,8 @@ currently the containerized build options are supported as follows:
```
Then you can pull and run the latest stable image using the below command:
```
docker pull mindspore/mindspore-gpu:1.0.1
docker run -it --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.0.1 /bin/bash
docker pull mindspore/mindspore-gpu:1.0.0
docker run -it -v /dev/shm:/dev/shm --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.0.0 /bin/bash
```
To test if the docker image works, please execute the python code below and check the output:

4
README_CN.md
View File

@ -168,8 +168,8 @@ MindSpore的Docker镜像托管在[Docker Hub](https://hub.docker.com/r/mindspore
```
使用以下命令获取并运行最新的稳定镜像：
```
docker pull mindspore/mindspore-gpu:1.0.1
docker run -it --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.0.1 /bin/bash
docker pull mindspore/mindspore-gpu:1.0.0
docker run -it -v /dev/shm:/dev/shm --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.0.0 /bin/bash
```
要测试Docker是否正常工作，请运行下面的Python代码并检查输出：