mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix dali installation to support paddle dev that has removed fluid (#2971)

pull/2975/head
Tingquan Gao 2023-09-15 17:38:24 +08:00 committed by GitHub
parent ef796960c0
commit 75cbc51b7f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test_tipc/prepare.sh
View File

@ -179,7 +179,9 @@ fi
if [[ ${FILENAME} == *ultra* ]] || [[ ${FILENAME} == *dali* ]]; then
python_name=$(func_parser_value "${lines[2]}")
${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
# only nightly dali support paddle develop version because the fluid has been removed in paddle develop.
# ${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
${python_name} -m pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly --upgrade nvidia-dali-nightly-cuda110
fi
if [[ ${MODE} = "lite_train_lite_infer" ]] || [[ ${MODE} = "lite_train_whole_infer" ]]; then