Install CUDA 11 on Jetson Nano and Xavier NX

CUDA 11 has a wealth of features – from platform system software to everything you need to get started and develop GPU-accelerated applications. Here is an overview of the key software features in this release.

  • Support for the NVIDIA Ampere GPU architecture, including the new NVIDIA A100 GPU for accelerated scaling and shrinking of AI and HPC data centers; multi-GPU systems with NVSwitch architecture, such as the DGX A100 and HGX A100.
  • Multi-instance GPU (MIG) partitioning capabilities that are particularly beneficial to cloud service providers (CSPs) to improve GPU utilization.
  • New third-generation tensor cores that accelerate mixed-precision, matrix computing for different data types, including TF32 and Bfloat16.
  • Programming and APIs for task graphs, asynchronous data movement, fine-grained synchronization, and secondary cache-resident control.
  • Performance optimizations for linear algebra, FFT, and matrix multiplication in the CUDA library.
  • Updates to the Nsight product family of tools for tracing, profiling, and debugging CUDA applications.
  • Full support for all major CPU architectures, including x86_64, Arm64 server, and POWER architectures.

The installation process on the Jetson platform is also quite simple. Copy the following code into your terminal then execute.

  1. wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/sbsa/cuda-ubuntu1804.pin
  2. sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
  3. wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda-repo-ubuntu1804-11-0-local_11.0.2-450.51.05-1_arm64.deb
  4. sudo dpkg -i cuda-repo-ubuntu1804-11-0-local_11.0.2-450.51.05-1_arm64.deb
  5. sudo apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub
  6. sudo apt-get update
  7. sudo apt-get -y install cuda

After installation, run command

dpkg -l | grep cuda

Then you’ll see CUDA 11 is installed.

About Author

Calendar

July 2020
M T W T F S S
 12345
6789101112
13141516171819
20212223242526
2728293031