Install CUDA, cuDNN ,TensorRT on RTX4060 NVIDIA GPU for Machine Learning AI

Image by Nana Dua

Install CUDA, cuDNN ,TensorRT on RTX4060 NVIDIA GPU for Machine Learning AI

I've new computer notebook then I want to use with Machine Learning AI.

My Computer Notebook Spec.

  • CPU: Intel Core i5-12450H (2.00GHz up to 4.40GHz, 8C(4P+4E)/12T, 12MB Intel Smart Cache)
  • RAM: 16GB (2x8GB) DDR4-3200 SO-DIMM
  • SSD 512GB NVMe PCIe 4.0
  • Graphics: NVIDIA GeForce RTX 4060 Laptop 8GB GDDR6
  • Windows 11 Home Edition




I want to use Dual-boot Windows and Ubuntu for Machine Learning AI Application.

Install CUDA, cuDNN ,TensorRT for Ubuntu

First Install Ubuntu alongside Windows

Step 1 : Create Ubuntu USB Drive Boot

We decide to use Ubuntu 22.04.3 LTS then load Ubuntu https://ubuntu.com/download/desktop


Step 2 : Boot with USB Drive then install Ubuntu


Step 3 : Select NVIDIA Driver

I've found nvidia-driver-525 and nvidia-driver-535 work fine with my notebook.

nvidia-driver-545 is not work with RTX4060.


  • Compatible NVIDIA GPU: Verify that your NVIDIA GPU is supported by the desired CUDA version you plan to install. Consult the CUDA Toolkit Compatibility Notice for guidance: https://docs.nvidia.com/deeplearning/frameworks/support-matrix/

    • check with nvidia-smi

      Step 4 : Install CUDA

      we select CUDA 12.0


      check with nvcc --version

      CUDA Install guide

      https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html


      Step 5 : Install cuDNN



      cuDNN Install guide

      https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html


      Step 6 : Install TensorRT

      Some Application use TensorRT for better performance.

      Download TensorRT

      https://developer.nvidia.com/tensorrt



      TensorRT install guide

      https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html






      Install CUDA, CUDNN ,TensorRT for Windows11

      STEP  1 : First check NVIDIA GPU Driver. 

      Normally your notebook have installed. 

      STEP 2 : Install CUDA 

       https://developer.nvidia.com/cuda-downloads

      We use CUDA 11.8 only.


      Install from EXE file. It has many CUDA program to install.


      STEP 3 : Install cuDNN

      Download file https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/


      Select cuDNN version as you want. we use cuDNN 8.9.0 CUDA 11

      Install Documents

      https://docs.nvidia.com/deeplearning/cudnn/installation/windows.html

      https://docs.nvidia.com/deeplearning/cudnn/installation/windows.html#installing-on-windows




      STEP 4 : Install TensorRT

      Install Guide
      https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html


      Now ( Feb 2024 ) TensorRT support up to CUDA 12.1 and cuDNN 8.9.0 only.

      Download TensorRT https://developer.nvidia.com/tensorrt-download


      Document Guide for Windows


      pip list and import tensorrt







      Adun Nantakaew อดุลย์ นันทะแก้ว
      LINE : adunnan


      ความคิดเห็น

      โพสต์ยอดนิยมจากบล็อกนี้

      Llama3 Typhoon v1.5 (scb10x) LLM

      Eval Llama v3.1 8B ,70B Model with Thai Language

      LLAMA-CPP-PYTHON on RTX4060 GPU