diff --git a/package_specific.sh b/package_specific.sh index d6465d6..a805929 100644 --- a/package_specific.sh +++ b/package_specific.sh @@ -48,3 +48,9 @@ if [[ $REPO == "facebookresearch/fairseq" ]]; then pip install cython patch -p0 < "$SCRIPT_DIR"/package_specific/fairseq_cub.patch fi + +if [[ $REPO == "open-mmlab/mmcv" ]] \ + && [[ $TORCH_VERSION == "1.12.1" ]] \ + && [[ $COMPUTE_PLATFORM == "cu102" ]]; then + patch -p0 < "$SCRIPT_DIR"/package_specific/mmcv_cpp14.patch +fi diff --git a/package_specific/mmcv_cpp14.patch b/package_specific/mmcv_cpp14.patch new file mode 100644 index 0000000..ebeecf0 --- /dev/null +++ b/package_specific/mmcv_cpp14.patch @@ -0,0 +1,8 @@ +--- setup.py ++++ setup.py +@@ -204 +204 @@ +- if parse_version(torch.__version__) <= parse_version('1.12.1'): ++ if parse_version(torch.__version__) <= parse_version('1.12.2'): +@@ -441 +441 @@ +- if parse_version(torch.__version__) <= parse_version('1.12.1'): ++ if parse_version(torch.__version__) <= parse_version('1.12.2'):