diff --git a/requirements_3.8.txt b/requirements_3.8.txt new file mode 100644 index 0000000..3650b5c --- /dev/null +++ b/requirements_3.8.txt @@ -0,0 +1,10 @@ +numpy==1.19.5 +opencv-python==4.1.2.30 +scipy==1.4.1 +tensorflow==2.5.0 +colorama==0.4.3 +tqdm==4.47.0 +ffmpeg-python==0.2.0 +Pillow==7.2.0 +scikit-image==0.17.2 +h5py==3.1.0 diff --git a/scripts/0_setup.sh b/scripts/0_setup.sh index 4100904..6191c9a 100755 --- a/scripts/0_setup.sh +++ b/scripts/0_setup.sh @@ -15,6 +15,14 @@ fi source .dfl/env/bin/activate -pip install -r requirements.txt +version=$(python -V | cut -f 2 -d ' ' | cut -f 1,2 -d .) +reqs_file='requirements.txt' + +if [[ ! -z "$version" && -f "requirements_$version.txt" ]]; then + reqs_file="requirements_$version.txt" +fi + +echo "Using $reqs_file" +pip install -r $reqs_file echo "Done."