CUDA_VERSION:

CUDNN_VERSION:

WITH_XPU_BKCL: "ON"

WITH_XPU_XRE5: "ON"

CACHE_DIR: /root/.cache

CCACHE_DIR: /root/.ccache

echo "Training Resnet50 completed!"

#inference Reset50

echo "Starting to predict ResNet50 model..."

python main.py -c paddlex/configs/modules/image_classification/ResNet50.yaml \

-o Global.mode=predict \

-o Predict.model_dir="./resnet50_output/best_model/inference" \

-o Predict.input="https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_classification_001.jpg" \

echo "Predicting Resnet50 completed!"

cd ..

export FLAGS_enable_pir_api=1

echo "if you use setup.py to compile,please export envs as following in /paddle ..."

cat << EOF

echo "if you use cmake to compile,please Configuring cmake in /paddle/build ..."

cat <<EOF

export CMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE:-Release}

export WITH_IPU=${WITH_IPU:-OFF}

export XPU_SDK_ROOT=${XPU_SDK_ROOT:-""}

export WITH_XPU_BKCL=${WITH_XPU_BKCL:-OFF}

export WITH_XPU_XRE5=${WITH_XPU_XRE5:-OFF}

export WITH_ARM=${WITH_ARM:-OFF}

export WITH_STRIP=${WITH_STRIP:-ON}

export ON_INFER=${ON_INFER:-OFF}

-DWITH_IPU=${WITH_IPU:-OFF} \

-DXPU_SDK_ROOT=${XPU_SDK_ROOT:-""} \

-DWITH_XPU_BKCL=${WITH_XPU_BKCL:-OFF} \

-DWITH_ARM=${WITH_ARM:-OFF} \

-DWITH_STRIP=${WITH_STRIP:-ON} \

-DON_INFER=${ON_INFER:-OFF} \