Hands-On Deep Learning with Go

Linux

As we've discussed, once CUDA is set up nicely, running your Gorgonia code on your GPU is as simple as adding -tags=cuda when building it. But how do we get to a point where that is possible? Let's find out.

This guide requires you to install standard Ubuntu 18.04. NVIDIA provides distribution-independent instructions (and troubleshooting steps) at: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html.

At a high level, you need to install the following packages:

• NVIDIA driver
• CUDA
• cuDNN
• libcupti-dev

First, you need to ensure you have NVIDIA's proprietary (not the open source default) driver installed. A quick way to check whether you are running it is to execute nvidia-smi. You should see output similar to the following, ...