diff --git a/requirements-gpu.txt b/requirements-gpu.txt
index 75dd998..f5bd746 100644
--- a/requirements-gpu.txt
+++ b/requirements-gpu.txt
@@ -5,7 +5,8 @@ onnx>=1.14.0
 protobuf>=3.20.2
 opencv-python
 pandas
+albumentations==1.3.1
 pydantic
 safetensors
 onnxruntime>=1.15.0
-onnxruntime-gpu>=1.15.0
\ No newline at end of file
+onnxruntime-gpu>=1.15.0