Browse Source

fix nccl bug in gpu docker container

tags/v1.1.0
wuweikang 5 years ago
parent
commit
64c4b805ce
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      README.md
  2. +1
    -1
      README_CN.md

+ 1
- 1
README.md View File

@@ -173,7 +173,7 @@ 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.0
docker run -it --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.0.0 /bin/bash
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:


+ 1
- 1
README_CN.md View File

@@ -169,7 +169,7 @@ MindSpore的Docker镜像托管在[Docker Hub](https://hub.docker.com/r/mindspore
使用以下命令获取并运行最新的稳定镜像:
```
docker pull mindspore/mindspore-gpu:1.0.0
docker run -it --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.0.0 /bin/bash
docker run -it -v /dev/shm:/dev/shm --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.0.0 /bin/bash
```

要测试Docker是否正常工作,请运行下面的Python代码并检查输出:


Loading…
Cancel
Save