diff --git a/python/requirements_ml_docker.txt b/python/requirements_ml_docker.txt
index cb975276e..73a3480c1 100644
--- a/python/requirements_ml_docker.txt
+++ b/python/requirements_ml_docker.txt
@@ -1,5 +1,5 @@
 ipython
-tensorflow-gpu>=2.4.0
+tensorflow-gpu==2.4.2
 -f https://download.pytorch.org/whl/torch_stable.html
 torch==1.8.1+cu111
 -f https://download.pytorch.org/whl/torch_stable.html
diff --git a/release/RELEASE_CHECKLIST.md b/release/RELEASE_CHECKLIST.md
index a23f1d1ff..86a40d8cb 100644
--- a/release/RELEASE_CHECKLIST.md
+++ b/release/RELEASE_CHECKLIST.md
@@ -82,6 +82,7 @@ This checklist is meant to be used in conjunction with the RELEASE_PROCESS.rst d
     - [ ] ft_small_elastic
 
 ## Final Steps
+- [ ] ML Docker Image Updated
 - [ ] Wheels uploaded to Test PyPI
 - [ ] Wheel sanity checks with Test PyPI
 	- [ ] Windows
diff --git a/release/RELEASE_PROCESS.rst b/release/RELEASE_PROCESS.rst
index 2e839fda4..a9d5dc438 100644
--- a/release/RELEASE_PROCESS.rst
+++ b/release/RELEASE_PROCESS.rst
@@ -309,6 +309,11 @@ to proceed with the final stages of the release!
    of the docs, trigger a new build of the "latest" branch in
    readthedocs to see if that fixes it.
 
+7. **Update ML Docker Image:** Upgrade the ``requirements_ml_docker.txt`` dependencies to use the same Tensorflow version as
+``requirements.txt`` and Torch version as minimum of ``requirements_tune.txt`` and ``requirements_rllib.txt``. Make any changes to the CUDA
+version so that it is compatible with these Tensorflow (https://www.tensorflow.org/install/source#gpu) or Torch (https://pytorch.org/get-started/locally/, https://pytorch.org/get-started/previous-versions/)
+versions. If there is no CUDA version that is compatible, then downgrade either Tensorflow or Torch. Ping @ijrsvt or @amogkam for assistance.
+
 7. **Update latest Docker Image:** SET THE VERSION NUMBER IN `docker/fix-docker-latest.sh`, then run the script ot update the "latest" tag
    in Dockerhub for the 
    ``rayproject/ray`` and ``rayproject/ray-ml`` Docker images to point to the Docker images built from the release. (Make sure there is no permission denied error, you will likely have to ask Thomas for permissions).