Skip to content

Commit 6c200e9

Browse files
committed
Fix model unittest CI
1 parent 950ac99 commit 6c200e9

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

.github/workflows/model-unittest-gpu.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ defaults:
3434
jobs:
3535
model-unittest-gpu-ci:
3636
name: model-unittest-gpu-ci
37-
runs-on: ${{ inputs.runner || 'ernie-8gpu' }}
37+
runs-on: ${{ inputs.runner || 'ernie-8gpu-1' }}
3838
steps:
3939
- name: Determine Image Name
4040
env:

scripts/regression/ci_model_unittest.sh

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ install_requirements() {
3030
python -m pip install -r requirements-dev.txt
3131
python -m pip install -r tests/requirements.txt
3232
python -m pip uninstall paddlepaddle paddlepaddle_gpu -y
33-
python -m pip install --no-cache-dir ${paddle} --no-dependencies --progress-bar off
33+
python -m pip install --no-cache-dir ${paddle} --no-dependencies --progress-bar off --force-reinstall
3434
python -c "import paddle;print('paddle');print(paddle.__version__);print(paddle.version.show())" >> ${log_path}/commit_info.txt
3535

3636
python setup.py bdist_wheel > /dev/null
@@ -104,8 +104,9 @@ if [[ ${FLAGS_enable_CI} == "true" ]] || [[ ${FLAGS_enable_CE} == "true" ]];then
104104
echo ' Testing all model unittest cases '
105105
unset http_proxy && unset https_proxy
106106
set +e
107-
echo "Check Cuda Version"
107+
echo "Check paddle Cuda Version"
108108
python -c "import paddle; print(paddle.version.cuda()); print(paddle.version.cudnn()); print(paddle.is_compiled_with_cuda())"
109+
echo "Check docker Cuda Version"
109110
nvcc -V
110111
cat /usr/local/cuda/version.txt
111112
PYTHONPATH=$(pwd) \

0 commit comments

Comments
 (0)