Update README.md

3 years ago · f0a891744d
parent 3a59e60b29
commit f0a891744d
1 changed files with 2 additions and 1 deletions
--- 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
 ```