How do I use my [AMD] GPU

[NiuAWSM]

Hello!
Im having a blast training the AI! :D
Im running the training on win11 right now. It works, but my CPU is so much worse than my GPU...
What do I have to do to run it on my 7900xtx?
I followed the win guide to setup the project.
Help is much appreciated^^ :)
Thanks!

[PWhiddy]

Hi!

Generally ML frameworks like pytorch (which stable baselines and therefore this repo uses) don't have great support for non-nvidia gpus.

I'm not sure if using an amd gpu with pytorch on windows as possible at all. You might be able to get it to work by using windows subsystem for linux (WSL) and installing the ROCm version of pytorch, but it might be a bit of a process and I have not tested it myself.

Hope that's helpful! Do report back if you end up trying it :)

[NiuAWSM]

Hey thanks for answering!
I got pytorch 2.1.2+rocm6.1.3 working on WSL.
how do I use GPU instead of CPU?

[PWhiddy]

For others seeing this issue, following these instructions seemed to do the trick

https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/wsl/howto_wsl.html

if this is installed correctly the gpu should be used automatically