diff --git a/packages/requirements-gpu.txt b/packages/requirements-gpu.txt index c485884..989a4e9 100644 --- a/packages/requirements-gpu.txt +++ b/packages/requirements-gpu.txt @@ -1,4 +1,4 @@ -tensorflow-gpu==1.14.0 +tensorflow-gpu==2.12.0 torch==1.9.0 torchvision==0.10.0 mxnet-cu101==1.5.0