Skip to content

Add MXNet GPU + GluonNLP and GluonCV toolkits#484

Merged
rosbo merged 3 commits intoKaggle:masterfrom
Ishitori:master
Feb 25, 2019
Merged

Add MXNet GPU + GluonNLP and GluonCV toolkits#484
rosbo merged 3 commits intoKaggle:masterfrom
Ishitori:master

Conversation

@Ishitori
Copy link

This pull request adds GPU version of MXNet to GPU kernel + GluonNLP and GluonCV toolkits to both kernels.

I use pure (not MKL version) of MXNet in case MKL is not supported on the host.
I use CUDA 9.2 version, as I see other packages of GPU kernel relies on 9.2 version of CUDA. Feel free to change to MKL versions if it is guaranteed to present on the host: mxnet-mkl and mxnet-cu92mkl packages for CPU and GPU respectively.

Relates to #452
@rosbo, thanks for fixing the master!

@rosbo rosbo self-requested a review February 20, 2019 23:11
@rosbo
Copy link
Contributor

rosbo commented Feb 21, 2019

Hi @Ishitori,

Can you add smoke tests for GluonNLP + CV and also tests that exercises the GPU?

See the tensorflow tests for an example:
https://github.com/Kaggle/docker-python/blob/master/tests/test_tensorflow.py

For tests that requires a GPU, add the @gpu_test annotation. See example linked above.

Thank you

@rosbo rosbo self-assigned this Feb 21, 2019
@rosbo
Copy link
Contributor

rosbo commented Feb 25, 2019

Thank you @Ishitori for adding the tests.

@rosbo rosbo merged commit 8dc2d4e into Kaggle:master Feb 25, 2019
@rosbo
Copy link
Contributor

rosbo commented Mar 5, 2019

Hi @Ishitori,

We have released new docker python images yesterday. Your new packages are available on Kaggle Kernels :)

Thank you

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants