Skip to content
This repository has been archived by the owner on Dec 21, 2023. It is now read-only.

CUDA 9 and cuDNN 7 support #17

Closed
gen-ko opened this issue Dec 9, 2017 · 5 comments
Closed

CUDA 9 and cuDNN 7 support #17

gen-ko opened this issue Dec 9, 2017 · 5 comments
Assignees

Comments

@gen-ko
Copy link

gen-ko commented Dec 9, 2017

Since CUDA 9 and cuDNN 7 has already released, and there are multiple API changes for these low-level libraries. Something like cudnnGetConvolutionForwardAlgorithm_v7() may have better performance than older version. We should adapt to these as well.

@gustavla
Copy link
Collaborator

gustavla commented Dec 9, 2017

All CUDA code runs through MXNet, which is a deep learning framework. If MXNet supports CUDA 9, then so can we. The problem is that they have not put up a wheel for mxnet-cu90==0.11.0, which is the only version we currently support (there are breaking changes in 0.12).

We hope to extend our MXNet version support soon, in which case we should also get CUDA 9 support for free. I'm not sure what cuDNN they currently support.

@srikris
Copy link
Contributor

srikris commented Dec 9, 2017

It looks like mxnet 0.12 supports CUDA 7 and 9 so we should try and depend on that version in the next release (of course assuming all our tests pass and the model quality stays the same).

Thanks for the feedback @linthieda!

@helloniklas
Copy link

@gustavla @srikris So I managed to compile from source and run MXNET 1.0.1 on GPU on my MacBook Pro with a Sonnett eGPU and NVIDIA 1080 card.

Since NVIDIA's macOS 10.13.2 driver is only supporting CUDA 9 and not CUDA 8, CUDA 9 is essential for this. I'm also running cuDNN 7 with this setup.

With Turi being an Apple thing and supporting CoreML export for Xcode presumably a lot of users will be on Mac. Would be great if next version will use MXNET 1.0.0 as you will then get GPU support for Mac with NVIDIA cards.

@gustavla
Copy link
Collaborator

gustavla commented Jan 3, 2018

@helloniklas We are close to both MXNet 0.12 and 1.0 support. However, the 1.0.0 version causes the object detector to crash (this is fixed in their 1.0.1 nightlies), so we will have to wait until 1.0.1 has been released before we can extend official support to 1.*. Hopefully they release it before our next version so that we can get it in. Thanks for your patience!

@gustavla
Copy link
Collaborator

gustavla commented Jan 5, 2018

All changes to support all newer versions of MXNet (which include CUDA 9 from 0.12.1) have been merged in (#129, #164). Unless you do a from source installation, this will all be available in the next release.

@gustavla gustavla closed this as completed Jan 5, 2018
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants