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.