Update Dockerfile to use CMake-based build (#1630)
* [DOCKER] Update Dockerfile to switch to CMake build * Fix typo in Dockerfile.ubuntu * Add README to docker sub-dir * Update Dockerfile and README * Modify markdown headings in docker/README * Update docs * Fix typo in docs * Update docs/install/docker-install.rst Co-authored-by: Jeffrey Novotny <jnovotny@amd.com> * Update docs/install/docker-install.rst Co-authored-by: Jeffrey Novotny <jnovotny@amd.com> * Update docs/install/docker-install.rst Co-authored-by: Jeffrey Novotny <jnovotny@amd.com> * Update docker/README --------- Signed-off-by: nileshnegi <Nilesh.Negi@amd.com> Co-authored-by: Jeffrey Novotny <jnovotny@amd.com>
This commit is contained in:
committed by
GitHub
orang tua
1786c0268b
melakukan
bd1a5b38b6
+1
-27
@@ -99,33 +99,7 @@ RCCL package install requires sudo/root access because it installs under `/opt/r
|
||||
|
||||
## Docker build
|
||||
|
||||
Assuming you have docker installed on your system:
|
||||
|
||||
#### To build the docker image :
|
||||
|
||||
By default, the given Dockerfile uses `docker.io/rocm/dev-ubuntu-22.04:latest` as the base docker image, and then installs RCCL (develop branch) and RCCL-Tests (develop branch).
|
||||
```shell
|
||||
$ docker build -t rccl-tests -f Dockerfile.ubuntu --pull .
|
||||
```
|
||||
|
||||
The base docker image, rccl repo, and rccl-tests repo can be modified using `--build-args` in the `docker build` command above. E.g., to use a different base docker image:
|
||||
```shell
|
||||
$ docker build -t rccl-tests -f Dockerfile.ubuntu --build-arg="ROCM_IMAGE_NAME=rocm/dev-ubuntu-20.04" --build-arg="ROCM_IMAGE_TAG=6.2" --pull .
|
||||
```
|
||||
|
||||
#### To start an interactive docker container on a system with AMD GPUs :
|
||||
|
||||
```shell
|
||||
$ docker run -it --rm --device=/dev/kfd --device=/dev/dri --group-add video --ipc=host --network=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined rccl-tests /bin/bash
|
||||
```
|
||||
|
||||
#### To run rccl-tests (all_reduce_perf) on 8 AMD GPUs (inside the docker container) :
|
||||
|
||||
```shell
|
||||
$ mpirun --allow-run-as-root -np 8 --mca pml ucx --mca btl ^openib -x NCCL_DEBUG=VERSION /workspace/rccl-tests/build/all_reduce_perf -b 1 -e 16G -f 2 -g 1
|
||||
```
|
||||
|
||||
For more information on rccl-tests options, refer to the [Usage](https://github.com/ROCm/rccl-tests#usage) section of rccl-tests.
|
||||
Refer to [docker/README.md](docker/README.md "docker/README.md")
|
||||
|
||||
## Tests
|
||||
|
||||
|
||||
Reference in New Issue
Block a user