mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #12748 from ROCmSoftwarePlatform:fixedROCm_builds_typo

Browse Source 
PiperOrigin-RevId: 480445232
This commit is contained in:
jax authors 2022-10-11 13:54:30 -07:00
commit 7a323b7b9d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
build/rocm/README.md
View File

@ -9,9 +9,9 @@ This directory contains files and setup instructions t0 build and test JAX for R
./build/rocm/ci_build.sh --keep_image bash -c "./build/rocm/build_rocm.sh"
3. Launch a container: If the build was successful, there should be a docker image with name "jax-rocm:latest" in list of docker images (use "docker images" command to list them).
3. Launch a container: If the build was successful, there should be a docker image with name "jax_ci.rocm" in list of docker images (use "docker images" command to list them).
```
sudo docker run -it --device=/dev/kfd --device=/dev/dri --security-opt seccomp=unconfined --group-add video --entrypoint /bin/bash jax-rocm:latest
sudo docker run -it --device=/dev/kfd --device=/dev/dri --security-opt seccomp=unconfined --group-add video --entrypoint /bin/bash jax_ci.rocm:latest
```
***