TorchRL cpu-only installation

Dear all,
I was wondering if installing TorchRL for CPU-only usage is possible.

Regards,

I don’t think torchrl ships with custom CUDA kernels (please correct me if I’m wrong) and depends on the PyTorch installation for its execution. So yes, install the CPU-only PyTorch binary, torchrl afterwards, and it should work.

1 Like

As @ptrblck mentioned, nothing in torchrl requires you from having a GPU. You can confidently use it on cpu only.

1 Like

Hi everyone,

Many thanks for all the replies.

I was wondering if anyone could provide me with a simple “Hello World” example using TorchRL to help me get started.

Best regards,
P.

Check out the getting started!
https://pytorch.org/rl/stable/tutorials/getting-started-0.html