From 6cd69b90ddb212723f325f5ee0a081a13e6032d9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=8D=AB=E8=8B=8F?= Date: Wed, 13 Nov 2024 11:43:29 +0800 Subject: [PATCH] add tensorflow-probability to resource tar --- pai_jobs/deploy_ext.sh | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/pai_jobs/deploy_ext.sh b/pai_jobs/deploy_ext.sh index 5e54b2d5b..3c8383439 100755 --- a/pai_jobs/deploy_ext.sh +++ b/pai_jobs/deploy_ext.sh @@ -145,18 +145,23 @@ fi if [ ! -d "tensorflow_probability" ] then + if [ $is_tf15 -gt 0 ]; then + tfp_version='0.8.0' + else + tfp_version='0.5.0' + fi if [ ! -e "tensorflow_probability" ] then - wget http://easyrec.oss-cn-beijing.aliyuncs.com/3rdparty/probability-0.5.0.tar.gz + wget http://easyrec.oss-cn-beijing.aliyuncs.com/3rdparty/probability-${tfp_version}.tar.gz if [ $? -ne 0 ] then echo "tensorflow_probability download failed." fi fi - tar -xzvf probability-0.5.0.tar.gz --strip-components=1 probability-0.5.0/tensorflow_probability + tar -xzvf probability-${tfp_version}.tar.gz --strip-components=1 probability-${tfp_version}/tensorflow_probability rm -rf tensorflow_probability/examples rm -rf tensorflow_probability/g3doc - rm -rf probability-0.5.0.tar.gz + rm -rf probability-${tfp_version}.tar.gz fi tar -cvzhf $RES_PATH easy_rec datahub lz4 cprotobuf kafka faiss tensorflow_probability run.py