Adjust the devices:driver setting when using image tag 'rocm'?

[trharmon0]

Howdy. I have an AMD GPU and so I'm using the 'rocm' image. I've not been able to get it to run because it doesn't like the "devices:driver: cuda" setting. That makes sense. But I've also not been able to figure out the proper setting. "amd"? "rocm"?

thx

[YanWenKun]

For Linux host:

• https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/install/installrad/native_linux/install-pytorch.html

  --device=/dev/kfd \
  --device=/dev/dri \

For WSL2 host:

• https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/install/installrad/wsl/install-pytorch.html

--device=/dev/dxg \
-v /usr/lib/wsl/lib/libdxcore.so:/usr/lib/libdxcore.so \
-v /opt/rocm/lib/libhsa-runtime64.so.1:/opt/rocm/lib/libhsa-runtime64.so.1  \

I also updated the doc in rocm, make it shorter and cleaner.

You can also check the compose file for reference.