diff --git a/README.md b/README.md
index 27fbfc5..c2f0ea2 100644
--- a/README.md
+++ b/README.md
@@ -11,8 +11,9 @@ cd Stripformer
 conda create -n Stripformer python=3.6
 source activate Stripformer
 conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
-pip install opencv-python tqdm pyyaml joblib glog scikit-image tensorboardX albumentations==1.1.0
+pip install opencv-python tqdm pyyaml joblib glog scikit-image tensorboardX albumentations
 pip install -U albumentations[imgaug]
+pip install albumentations==1.1.0
 ```