use lightgbm with cuda instead of gpu directly

djherbis · djherbis · commit a66a1f2d0564 · 2024-12-04T15:35:57.000Z
diff --git a/tests/test_lightgbm.py b/tests/test_lightgbm.py
@@ -49,7 +49,7 @@ def test_gpu(self):
             'bagging_freq': 5,
             'force_row_wise': True,
             'verbose': 1,
-            'device': 'gpu'
+            'device': 'cuda'
         }
         
         # Run only one round for faster test

Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@ def test_gpu(self):`
`49`	`49`	`'bagging_freq': 5,`
`50`	`50`	`'force_row_wise': True,`
`51`	`51`	`'verbose': 1,`
`52`		`- 'device': 'gpu'`
	`52`	`+ 'device': 'cuda'`
`53`	`53`	`}`
`54`	`54`
`55`	`55`	`# Run only one round for faster test`