Skip to content

Commit 64bb588

Browse files
committed
Fix model unittest CI
1 parent 2c335c0 commit 64bb588

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

scripts/regression/ci_model_unittest.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,8 @@ if [[ ${FLAGS_enable_CI} == "true" ]] || [[ ${FLAGS_enable_CE} == "true" ]];then
111111
cat /usr/local/cuda/version.txt
112112
echo "Check nvidia-smi"
113113
nvidia-smi
114-
python -c "import paddle; print(paddle.device.get_device_count())"
114+
python -c "import paddle; print(paddle.device.device_count())"
115+
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
115116
PYTHONPATH=$(pwd) \
116117
COVERAGE_SOURCE=paddleformers \
117118
python -m pytest -s -v ${model_unittest_path} > ${log_path}/model_unittest.log 2>&1

0 commit comments

Comments
 (0)