Giters

ai4co / rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

Home Page:https://rl4.co

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[Notice] Updating to new TorchRL and TensorDict

fedebotu opened this issue · comments

commented

The new TorchRL and TensorDict versions have been officially released! We will be able to use Python 3.11 as well 🚀

However, there are some backward compatibility issues with the current RL4CO version and the new TorchRL. Moreover, it looks like the newest version needs PyTorch 2.1 to work for the time being(issue here, linking this one as well).

In the following couple of weeks, we will work towards the next release to address new compatibilities and making RL4CO more efficient ! Stay tuned

For the time being

To install RL4CO now, make sure TorchRL and Tensordict are at v0.1.1 (for Windows: PyTorch should be <2.1 for this version of TorchRL):

pip install torchrl==0.1.1 tensordict==0.1.1 rl4co==0.2.3
commented

Solved with the latest release 🚀