Skip to Content
Deep Learning
book

Deep Learning

by Josh Patterson, Adam Gibson
August 2017
Intermediate to advanced content levelIntermediate to advanced
530 pages
13h 23m
English
O'Reilly Media, Inc.
Content preview from Deep Learning

Appendix I. Setting Up GPUs for DL4J Projects

Training neural networks involves a host of linear algebra calculations. Graphics processing units (GPUs), with their thousands of cores, were designed to excel at such problems. They are therefore often used to speed up training and eventually attain better compute performance per dollar, per watt.

Switching Backends to GPU

DL4J can be used with nVIDIA GPUs with support currently for Cuda 7.5.  Cuda 8.0 will be supported when available. DL4J is written to be plug-and-play. This means that switching the compute layer from CPU to GPU is as easy as switching out the artifactId line under the  nd4j in the dependencies in the pom.xml file.

<dependencyManagement>
    <dependencies>
        <dependency>
            <groupId>org.nd4j</groupId>
            <artifactId>nd4j-cuda-7.5-platform</artifactId>
            <version>${nd4j.version}</version>
        </dependency>
    </dependencies>
</dependencyManagement>

Picking a GPU

In general, a high-end consumer-grade model or a professional Tesla device is recommended: as of this writing, a pair of NVidia GeForce GTX 1070 GPUs would be a solid choice with which to begin.

Here are features to consider when buying a GPU:

Amount of multiprocessors (or cores) on board

The more, the better. As simple as that. This relates to how many parallel threads the GPU can process at any given moment.

Memory available on device

This defines how much data can be uploaded ...

Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.
Start your free trial

You might also like

Deep Learning

Deep Learning

Andrew Glassner
Grokking Deep Learning

Grokking Deep Learning

Andrew W. Trask
Deep Learning with PyTorch

Deep Learning with PyTorch

Eli Stevens, Thomas Viehmann, Luca Pietro Giovanni Antiga

Publisher Resources

ISBN: 9781491924570Errata PageSupplemental Content