Issues with AMD Ryzen AI 7 350 w/ Radeon 860M, ROCm drivers

#9
by vmss2009 - opened

I am trying to train SO-101 using ASUS mini PC with AMD Ryzen AI 7 350 w/ Radeon 860M, 64 GB Memory.
I installed ROCm but could not find right drivers for "AMD Ryzen AI 7 350 w/ Radeon 860M". Due to this my training is defaulting to CPU instead of GPU.

This is what I have done till now -

Set up conda env (followed https://huggingface.co/docs/lerobot/installation)
Calibrated the leader and follower arms, teleoperation was successful
Recorded 50 episodes

Sign up or log in to comment