CVE-2021-41212: TensorFlow GPU - Out-of-Bounds Read in Shape Inference Code
Summary
CVE-2021-41212: TensorFlow is an open source platform for machine learning. Affects tensorflow-gpu across 5 version ranges.
< 2.4.4
>= 2.5.0, < 2.5.2
>= 2.6.0, < 2.6.1
>= 2.7.0rc0, < 2.7.0
< fa6b7782fbb14aa08d767bc799c531f5e1fb3bb8
Fixed in 4 releases; the minimum fixed version depends on the release line you run. How to fix ↓
Also tracked for
- tensorflow · Bitnami
- tensorflow · PyPI
- tensorflow · PyPI
- tensorflow-cpu · PyPI
The same CVE-2021-41212 is tracked separately for each ecosystem and vendor. Its fix information is the same; the package and affected versions differ.
Check if this vulnerability affects your application
Upload a manifest or SBOM and we'll show whether you're affected, plus every other risky package in the same scan.
No signup required · Results in about 60 seconds
Background
TensorFlow is an open-source platform for machine learning. It provides a comprehensive ecosystem of tools, libraries, and resources for developing and deploying machine learning models. TensorFlow GPU is a variant of TensorFlow that leverages GPU acceleration for faster computation. It is widely used in various industries and research domains for deep learning tasks.
Vulnerability Detail
The vulnerability (CVE-2021-41212) exists in the shape inference code for tf.ragged.cross in TensorFlow. It can trigger a read outside the bounds of a heap-allocated array. This can lead to memory corruption, potentially resulting in crashes, information disclosure, or even remote code execution. The issue affects multiple versions of TensorFlow, including 2.4.4, 2.5.2, 2.6.1, and 2.7.0rc0.
How To Fix
Latest published version: 2.12.0 as of 22 Oct 2026
pip install --upgrade "tensorflow-gpu==2.12.0"Upgrading to it may include breaking changes; the versions below are the minimum fixed release for this vulnerability.
If you can't move to the latest
The minimum fixed version depends on the release line you run.
-
< 2.4.42.4.4pip install --upgrade "tensorflow-gpu==2.4.4" -
>= 2.5.0, < 2.5.22.5.2pip install --upgrade "tensorflow-gpu==2.5.2" -
>= 2.6.0, < 2.6.12.6.1pip install --upgrade "tensorflow-gpu==2.6.1" -
>= 2.7.0rc0, < 2.7.02.7.0pip install --upgrade "tensorflow-gpu==2.7.0"
Tip — Not sure which release line you're on? Vulert can find the fix for the version you are using. Scan free →
Workaround
There is no specific workaround provided for this vulnerability. It is recommended to apply the fix by updating TensorFlow GPU to the patched version.
Conclusion
The vulnerability (CVE-2021-41212) in TensorFlow GPU's shape inference code for tf.ragged.cross can lead to memory corruption and potential security risks. It is crucial to update to the patched versions or upgrade to TensorFlow GPU 2.7.0 to mitigate the vulnerability. For further support and assistance in vulnerability remediation, please contact DevNack.com.
See every vulnerability recorded for tensorflow-gpu →
FAQ
TensorFlow GPU is a variant of TensorFlow that utilizes GPU acceleration for faster machine learning computations.
The vulnerability can lead to memory corruption, potentially resulting in crashes, information disclosure, or remote code execution.
The vulnerability affects TensorFlow GPU versions 2.4.4, 2.5.2, 2.6.1, and 2.7.0rc0.
To fix the vulnerability, update TensorFlow GPU to version 2.7.0 or apply the respective patched versions for TensorFlow 2.6.1, 2.5.2, and 2.4.4.
No specific workaround is provided. It is recommended to update to the patched version.
References
Related vulnerabilities
Recent vulnerabilities recorded for tensorflow-gpu.
