提交 925ae7a5 编写于 作者: G gaotingquan 提交者: cuicheng01

install nvidia-dali-cuda11.0

上级 b2dc1788
......@@ -178,9 +178,9 @@ if [[ $model_name == *ShiTu* ]]; then
fi
fi
if [[ $FILENAME == *use_dali* ]]; then
if [[ ${FILENAME} == *ultra* ]] || [[ ${FILENAME} == *dali* ]]; then
python_name=$(func_parser_value "${lines[2]}")
${python_name} -m pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly --upgrade nvidia-dali-nightly-cuda102
${python_name} -m pip install https://paddle-wheel.bj.bcebos.com/benchmark/nvidia_dali_cuda110-1.25.0-7922357-py3-none-manylinux2014_x86_64.whl
fi
if [[ ${MODE} = "lite_train_lite_infer" ]] || [[ ${MODE} = "lite_train_whole_infer" ]]; then
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册