diff --git a/setup.py b/setup.py index 238e6f3..b3a0c0a 100644 --- a/setup.py +++ b/setup.py @@ -8,7 +8,7 @@ packages=[package for package in find_packages() if package.startswith('rlcodebase')], install_requires=[ - 'torch==1.8.0', + 'torch==1.13.1', 'tensorflow>=2.5.1', 'gym==0.21.0', 'torchvision==0.9.0',