GithubHelp home page GithubHelp logo

CUDA 9 and cuDNN 7 support about turicreate HOT 5 CLOSED

apple avatar apple commented on June 23, 2024 1
CUDA 9 and cuDNN 7 support

from turicreate.

Comments (5)

gustavla avatar gustavla commented on June 23, 2024 3

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.

from turicreate.

gustavla avatar gustavla commented on June 23, 2024 2

@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!

from turicreate.

helloniklas avatar helloniklas commented on June 23, 2024 1

@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.

from turicreate.

gustavla avatar gustavla commented on June 23, 2024

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.

from turicreate.

srikris avatar srikris commented on June 23, 2024

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!

from turicreate.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.