File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -34,7 +34,7 @@ defaults:
34
34
jobs :
35
35
model-unittest-gpu-ci :
36
36
name : model-unittest-gpu-ci
37
- runs-on : ${{ inputs.runner || 'ernie-8gpu' }}
37
+ runs-on : ${{ inputs.runner || 'ernie-8gpu-1 ' }}
38
38
steps :
39
39
- name : Determine Image Name
40
40
env :
Original file line number Diff line number Diff line change @@ -30,7 +30,7 @@ install_requirements() {
30
30
python -m pip install -r requirements-dev.txt
31
31
python -m pip install -r tests/requirements.txt
32
32
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
34
34
python -c " import paddle;print('paddle');print(paddle.__version__);print(paddle.version.show())" >> ${log_path} /commit_info.txt
35
35
36
36
python setup.py bdist_wheel > /dev/null
@@ -104,8 +104,9 @@ if [[ ${FLAGS_enable_CI} == "true" ]] || [[ ${FLAGS_enable_CE} == "true" ]];then
104
104
echo ' Testing all model unittest cases '
105
105
unset http_proxy && unset https_proxy
106
106
set +e
107
- echo " Check Cuda Version"
107
+ echo " Check paddle Cuda Version"
108
108
python -c " import paddle; print(paddle.version.cuda()); print(paddle.version.cudnn()); print(paddle.is_compiled_with_cuda())"
109
+ echo " Check docker Cuda Version"
109
110
nvcc -V
110
111
cat /usr/local/cuda/version.txt
111
112
PYTHONPATH=$( pwd) \
You can’t perform that action at this time.
0 commit comments