Install NVIDIA Drivers and Vulkan on Ubuntu
Installation Prerequisites
Upgrade the system and the kernel, then the machine if needed.
Install Kernel Headers
Install kernel headers and development packages.
Install the extra packages.
Installing Vulkan Library
The rendering engine of HEAVY.AI (present in Enterprise Editions) requires a Vulkan-enabled driver and the Vulkan library. Without these components, the database itself may not be able to start.
Install the Vulkan library and its dependencies using apt
.
For more information about troubleshooting Vulkan, see the Vulkan Renderer section.
Installing NVIDIA Drivers
Installing NVIDIA drivers with support for the CUDA platform is required to run GPU-enabled versions of HEAVY.AI.
You can install NVIDIA drivers in multiple ways, we've outlined three available options below. If you would prefer not to decide, we recommend Option 1.
Option 1: Install NVIDIA drivers with CUDA toolkit from NVIDIA Website
Option 2: Install NVIDIA drivers via .run file using the NVIDIA Website
Option 3: Install NVIDIA drivers using APT package manager
It is advisable to keep a record of the installation method used, as upgrading NVIDIA drivers will require the utilization of the same method for successful results.
What is CUDA? What is the CUDA toolkit?
CUDA is a parallel computing platform and application programming interface (API) model. It uses a CUDA-enabled graphics processing unit (GPU) for general-purpose processing. The CUDA platform provides direct access to the GPU virtual instruction set and parallel computation elements. For more information on CUDA unrelated to installing HEAVY.AI, see https://developer.nvidia.com/cuda-zone.
The CUDA Toolkit from NVIDIA provides everything you need to develop GPU-accelerated applications. The CUDA Toolkit includes GPU-accelerated libraries, a compiler, development tools and the CUDA runtime. The CUDA Toolkit is not required to run HEAVY.AI, but you must install the CUDA toolkit if you use advanced features like C++ User-Defined Functions and or User-Defined Table Functions to extend the database capabilities.
Option 1: Install NVIDIA Drivers with CUDA Toolkit from NVIDIA Website
Open https://developer.nvidia.com/cuda-toolkit-archive and select the desired CUDA Toolkit version to install.
The minimum CUDA version supported by HEAVY.AI is 11.4. We recommend using a release that has been available for at least two months.
In the "Target Platform" section, follow these steps:
For "Operating System" select Linux
For Architecture" select x86_64
For "Distribution" select Ubuntu
For "Version" select the version of your operating system (20.04)
For "Installer Type" choose deb (network) **
One by one, run the presented commands in the Installer Instructions section on your server.
** You may optionally use any of the "Installer Type" options available.
If you choose to use the .run file option, prior to running the installer you will need to manually install build-essentials
using apt
and change permissions of the downloaded .run file to allow execution.
Option 2: Install NVIDIA Drivers via .run file using the NVIDIA Website
Install the CUDA package for your platform and operating system according to the instructions on the NVIDIA website (https://www.nvidia.com/download/index.aspx).
If you don't know the exact GPU model in your system run this command
You'll get an output in the format Product Type, Series and Model
In this example, the Product type is Tesla the Series is T (as Turing), and the model is T4.
Select the Product Type as the one you got with the command.
Select the correct Product Series and Product Type for your installation.
In the Operating System dropdown list, select Linux 64-bit.
In the CUDA Toolkit dropdown list, click a supported version (11.4 or higher).
Click Search.
On the resulting page, verify the download information and click Download
On the subsequent page, if you agree to the terms, right click on "Agree and Download" and select "Copy Link Address". You may also manually download and transfer to your server, skipping the next step.
On your server, type
wget
and paste the URL you copied in the previous step. Press enter to download.
Please check that the driver's version you are downloading meets the HEAVI.AI minimum requirements
Install the tools needed for installation.
Change the permissions of the downloaded .run file to allow execution, and run the installation.
Option 3: Install NVIDIA drivers using APT
Install a specific version of the driver for your GPU by installing the NVIDIA repository and using the apt
package manager.
Be careful when choosing the driver version to install. Ensure that your GPU's model is supported and that meets the HEAVI.AI minimum requirements
Run the command to get a list of the available driver's version
Install the driver version needed with apt
NVIDIA Driver Post-Installation steps
Reboot your system to ensure the new version of the driver is loaded
Verify Successful NVIDIA driver installation
Run nvidia-smi
to verify that your drivers are installed correctly and recognize the GPUs in your environment. Depending on your environment, you should see something like this to confirm that your NVIDIA GPUs and drivers are present.
![](https://docs.heavy.ai/~gitbook/image?url=https%3A%2F%2F604851066-files.gitbook.io%2F%7E%2Ffiles%2Fv0%2Fb%2Fgitbook-x-prod.appspot.com%2Fo%2Fspaces%252F-M6kHkWH17KJKscL0Cc4-887967055%252Fuploads%252Fqw4oBuX4cFFH2QVJzAZM%252Fimage.png%3Falt%3Dmedia%26token%3Deee93fbd-e4da-41ba-b6bb-1cd94e0c610e&width=768&dpr=4&quality=100&sign=8094fc5b&sv=1)
If you see an error like the following, the NVIDIA drivers are probably installed incorrectly:
Review the installation instructions, specifically checking for completion of install prerequisites, and correct any errors.
Install Vulkan library
The rendering engine of HEAVY.AI requires a Vulkan-enabled driver and the Vulkan library. Without these components, the database itself can't even start without disabling the back-end renderer.
Install the Vulkan library and its dependencies using apt
.
For more information about troubleshooting Vulkan, see the Vulkan Renderer section.
Advanced Installation
You must install the CUDA toolkit and Clang if you use advanced features like C++ User-Defined Functions and or User-Defined Table Functions to extend the database capabilities.
Install CUDA Toolkit ᴼᴾᵀᴵᴼᴺᴬᴸ
If you installed NVIDIA drivers using Option 1 above, the CUDA toolkit is already installed; you may proceed to the verification step below.
Install the NVIDIA public repository GPG key.
Add the repository.
List the available Cuda toolkit versions.
Install the CUDA toolkit using apt
.
Verification
Check that everything is working and the toolkit has been installed.
Install Clang ᴼᴾᵀᴵᴼᴺᴬᴸ
You must install Clang if you use advanced features like C++ User-Defined Functions and or User-Defined Table Functions to extend the database capabilities. Install Clang and LLVM dependencies using apt
.
Verification
Check that the software is installed and in the execution path.
For more information, see C++ User-Defined Functions.
Last updated