Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 39e153a

Browse files
Clarifying docker launch
1 parent 86bd6c7 commit 39e153a

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

examples/docker/README.md

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,13 @@ If you want to rebuild your image, and force the latest Apex to be cloned and in
2121
Currently, the non-`-devel` images on Pytorch Dockerhub do not contain the Cuda compiler `nvcc`. Therefore,
2222
images whose name does not contain `-devel` are not eligible candidates for `BASE_IMAGE`.
2323

24+
### Running your Apex container
25+
26+
Like any Cuda-enabled Pytorch container, a container with Apex should be run via [nvidia-docker](https://github.com/NVIDIA/nvidia-docker), for example:
27+
```
28+
docker run --runtime=nvidia -it --rm --ipc=host new_image_with_apex
29+
```
30+
2431
## Option 2: Install Apex in a running container
2532

2633
Instead of building a new container, it is also a viable option to `git clone https://github.com/NVIDIA/apex.git` on bare metal, mount the Apex repo into your container at launch by running, for example,

0 commit comments

Comments
 (0)