rail-berkeley / rlkit

Collection of reinforcement learning algorithms
MIT License
2.43k stars 547 forks source link

Upgrade PyTorch #127

Open xanderdunn opened 3 years ago

xanderdunn commented 3 years ago

At 0.4.1, the PytTorch version in the Anaconda environment .yml's is far behind the current stable PyTorch version 1.7.1.

hammer-wang commented 3 years ago

Have you tried to use rlkit with the current PyTorch version, would it work? @xanderdunn

xanderdunn commented 3 years ago

Yes, it did run for me when I upgraded to latest pytorch

Superyoutiao commented 2 years ago

Hi, did you modify other parts of the .yml file? For example the python version? It seems that python=3.5.2 doesn't work.