create a libnvidia-opencl.so.1 file after install lightgbm · JasonYangCode/docker-python@b2737cc · GitHub

JasonYangCode / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 1

• Code
• Pull requests
• Actions
• Projects
• Security and quality
• Insights

Additional navigation options

Commit b2737cc

Browse filesBrowse the repository at this point in the historyBrowse files

rosbo

committed

create a libnvidia-opencl.so.1 file after install lightgbm

1 parent 4cd57a8 commit b2737cc
Copy full SHA for b2737cc

2 files changed

+4-3Lines changed: 4 additions & 3 deletions

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• gpu.Dockerfile
• tests
  • test_lightgbm.py

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎gpu.Dockerfile‎

Copy file name to clipboardExpand all lines: gpu.Dockerfile

+2Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
@@ -61,6 +61,8 @@ RUN pip uninstall -y lightgbm && \
61	61	make -j$(nproc) && \
62	62	cd /usr/local/src/LightGBM/python-package && \
63	63	python setup.py install --precompile && \
	64	+ mkdir -p /etc/OpenCL/vendors && \
	65	+ echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd && \
64	66	/tmp/clean-layer.sh
65	67
66	68	# Install JAX

Collapse file

‎tests/test_lightgbm.py‎

Copy file name to clipboardExpand all lines: tests/test_lightgbm.py

+2-3Lines changed: 2 additions & 3 deletions

Original file line number	Diff line number	Diff line change
@@ -46,16 +46,15 @@ def test_gpu(self):
46	46	lgb_eval = lgb.Dataset(iris.data[100:], iris.target[100:], reference=lgb_train)
47	47
48	48	params = {
49		- 'task': 'train',
50	49	'boosting_type': 'gbdt',
51	50	'objective': 'regression',
52		- 'metric': {'l2', 'auc'},
	51	+ 'metric': 'auc',
53	52	'num_leaves': 31,
54	53	'learning_rate': 0.05,
55	54	'feature_fraction': 0.9,
56	55	'bagging_fraction': 0.8,
57	56	'bagging_freq': 5,
58		- 'verbose': 0,
	57	+ 'verbose': 1,
59	58	'device': 'gpu'
60	59	}
61	60

0 commit comments

Comments

0 (0)