Fix paddle ssd (#2765)

awolant · web-flow · commit cfffe9f62904 · 2021-03-07T09:29:43.000+01:00
Make Paddle SSD L1 pass on multi-gpu system.

Signed-off-by: Albert Wolant &lt;awolant@nvidia.com&gt;
diff --git a/qa/TL1_paddle_ssd/test.sh b/qa/TL1_paddle_ssd/test.sh
@@ -5,6 +5,9 @@ pip_packages="paddlepaddle-gpu"
 target_dir=./docs/examples/use_cases/paddle/ssd/
 
 test_body() {
+    # This test is incompatible with multi-gpu environments.
+    export CUDA_VISIBLE_DEVICES=0
+    
     python train.py --check-loss-steps=300 -b 8 /data/coco/coco-2017/coco2017/
 }
 

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,9 @@ pip_packages="paddlepaddle-gpu"`
`5`	`5`	`target_dir=./docs/examples/use_cases/paddle/ssd/`
`6`	`6`
`7`	`7`	`test_body() {`
	`8`	`+ # This test is incompatible with multi-gpu environments.`
	`9`	`+ export CUDA_VISIBLE_DEVICES=0`
	`10`	`+`
`8`	`11`	`python train.py --check-loss-steps=300 -b 8 /data/coco/coco-2017/coco2017/`
`9`	`12`	`}`
`10`	`13`