File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -116,10 +116,6 @@ jobs:
116
116
cp -r $work_dir/../../../models ./models
117
117
echo "Check models:"
118
118
ls -l ./models
119
- echo "Check Cuda Version"
120
- python -c "import paddle; print(paddle.version.cuda()); print(paddle.version.cudnn()); print(paddle.is_compiled_with_cuda())"
121
- nvcc -V
122
- cat /usr/local/cuda/version.txt
123
119
echo "Test Start"
124
120
timeout 30m bash scripts/regression/ci_model_unittest.sh ${paddle_whl}
125
121
'
Original file line number Diff line number Diff line change @@ -104,6 +104,10 @@ 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"
108
+ python -c " import paddle; print(paddle.version.cuda()); print(paddle.version.cudnn()); print(paddle.is_compiled_with_cuda())"
109
+ nvcc -V
110
+ cat /usr/local/cuda/version.txt
107
111
PYTHONPATH=$( pwd) \
108
112
COVERAGE_SOURCE=paddleformers \
109
113
python -m pytest -s -v ${model_unittest_path} > ${log_path} /model_unittest.log 2>&1
You can’t perform that action at this time.
0 commit comments