CVE-2021-37640: TensorFlow GPU - Division by 0 in SparseReshape Implementation
Summary
CVE-2021-37640: TensorFlow is an end-to-end open source platform for machine learning. Affects tensorflow-gpu across 3 version ranges.
>= 2.3.0, < 2.3.4
>= 2.4.0, < 2.4.3
< 4923de56ec94fff7770df259ab7f2288a74feb41
Fixed in 2.3.4 or 2.4.3, depending 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-37640 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 end-to-end 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 commonly used in scenarios where large-scale machine learning tasks require high-performance computing.
Vulnerability Detail
The vulnerability (CVE-2021-37640) exists in the implementation of tf.raw_ops.SparseReshape in TensorFlow GPU. The issue arises when the reshaping functor is called without checking if the input or target shape has a non-zero number of elements. As a result, an integral division by 0 exception can be triggered. This vulnerability can be exploited by an attacker to cause a denial-of-service (DoS) condition or potentially execute arbitrary code.
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.3.0, < 2.3.42.3.4pip install --upgrade "tensorflow-gpu==2.3.4" -
>= 2.4.0, < 2.4.32.4.3pip install --upgrade "tensorflow-gpu==2.4.3"
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 or update to the patched versions as mentioned in the 'how_to_fix' section.
Conclusion
The vulnerability in TensorFlow GPU (CVE-2021-37640) can result in a division by 0 exception, potentially leading to a denial-of-service condition or arbitrary code execution. It is crucial to update to the patched versions or apply the provided fix to mitigate the risk. For further support on vulnerability remediation, you can reach out to 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.
This vulnerability can lead to a division by 0 exception, potentially causing a denial-of-service (DoS) condition or enabling arbitrary code execution.
To fix this vulnerability, update TensorFlow GPU to version 2.6.0 or apply the patch included in GitHub commit 4923de56ec94fff7770df259ab7f2288a74feb41.
No specific workarounds have been provided. It is recommended to update to the patched versions or apply the fix.
While this vulnerability can potentially lead to arbitrary code execution, there is no direct mention of data loss or compromise in the vulnerability details.
References
Related vulnerabilities
Recent vulnerabilities recorded for tensorflow-gpu.
