diff --git a/requirements-gpu.txt b/requirements-gpu.txt index c4b7cd1..e918275 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -1,4 +1,4 @@ -tensorflow-gpu==2.3.1 +tensorflow-gpu==2.5.3 pyyaml>=5.3.1 gym>=0.17.1 stable-baselines==2.10 \ No newline at end of file