Search icon CANCEL
Arrow left icon
Explore Products
Best Sellers
New Releases
Books
Videos
Audiobooks
Learning Hub
Conferences
Free Learning
Arrow right icon
Arrow up icon
GO TO TOP
Hands-On GPU Computing with Python

You're reading from   Hands-On GPU Computing with Python Explore the capabilities of GPUs for solving high performance computational problems

Arrow left icon
Product type Paperback
Published in May 2019
Publisher Packt
ISBN-13 9781789341072
Length 452 pages
Edition 1st Edition
Languages
Tools
Arrow right icon
Author (1):
Arrow left icon
Avimanyu Bandyopadhyay Avimanyu Bandyopadhyay
Author Profile Icon Avimanyu Bandyopadhyay
Avimanyu Bandyopadhyay
Arrow right icon
View More author details
Toc

Table of Contents (17) Chapters Close

Preface 1. Section 1: Computing with GPUs Introduction, Fundamental Concepts, and Hardware
2. Introducing GPU Computing FREE CHAPTER 3. Designing a GPU Computing Strategy 4. Setting Up a GPU Computing Platform with NVIDIA and AMD 5. Section 2: Hands-On Development with GPU Programming
6. Fundamentals of GPU Programming 7. Setting Up Your Environment for GPU Programming 8. Working with CUDA and PyCUDA 9. Working with ROCm and PyOpenCL 10. Working with Anaconda, CuPy, and Numba for GPUs 11. Section 3: Containerization and Machine Learning with GPU-Powered Python
12. Containerization on GPU-Enabled Platforms 13. Accelerated Machine Learning on GPUs 14. GPU Acceleration for Scientific Applications Using DeepChem 15. Other Books You May Enjoy Appendix A

Installing TensorFlow and PyTorch for GPUs

Since we have already learned about CUDA's installation and implementation, it will now be easier for us to get started on our TensorFlow and PyTorch installation procedure. Additionally, we also need cuDNN to be installed, which is the predefined deep neural network library for CUDA. To be able to download the library, you have to fill in a free registration at https://developer.nvidia.com/, which is the official web portal for the NVIDIA Developer program.

Installing cuDNN

In this section, we are going to install cuDNN 7.4.2 for both TensorFlow and PyTorch. Follow these steps to get started:

  1. Download the corresponding archive from the cuDNN repository:
  1. Use a Terminal and...
lock icon The rest of the chapter is locked
Register for a free Packt account to unlock a world of extra content!
A free Packt account unlocks extra newsletters, articles, discounted offers, and much more. Start advancing your knowledge today.
Unlock this book and the full library FREE for 7 days
Get unlimited access to 7000+ expert-authored eBooks and videos courses covering every tech area you can think of
Renews at AU $24.99/month. Cancel anytime