ML Agents, what can I use for AMD?

[sleek rapids]

(full newbie with ML)So from what I've understood CUDA is for Nvidia, but does it work with AMD? If not, what should I use instead in combinationg with PyTorch?

[sly lagoon]

Pytorch doesn't have cuda as a requirement, it can run on AMD cards and even CPU.

And regardless, you don't need to install anything extra. Just the right pytorch version.

[sleek rapids]

Oh, fair enough. They mentioned installing it on the install guide so thought I needed it. Thanks a lot!

[sly lagoon]

If you install it via pip, you probably need to provide the right args or url for a version that supports amd gpus(which I assume you have) if you want it to run on the gpu.

[sleek rapids]

So what would be the simplest way for me to get this going? Ideally I want to just get up and running, quite a lot to pick up atm 😅

[sly lagoon]

Follow the installation guides of ml agents package for starters. You can worry about it using the gpu or not when you have it working.

[sleek rapids]

Sounds good. Miniconda being a bit of a struggle to get working but should be goochie

[sly lagoon]

Miniconda is just for virtual environment, to avoid dirtying your PC env. You could probably skip it or use venv.

[sleek rapids]

Okey, skipping that for now then. But you mentioned pip and extra steps? Would it be easier with libTorch or source instead?