CUDA Installation and GPU Stress

CUDA Installation

The easiest way is to install CUDA is by leveraging the Jetson SDKManager to install CUDA, CUDA-X AI, Computer Vision.

Steps

• Prepare a host computer with the latest SDKManager installed.

• Boot PCIe-NX150 into OS, and make sure the host computer and PCIe-NX150 inside a local network, and the host can connect to PCIe-NX150 by ssh.

  
• On host computer, open SDKManager, select Jetson Orin NX modules along with JetPack 5.1.1. Then, click on CONTINUE to STEP 02

  
• Under the TARGET COMPONENTS, uncheck Jetson Linux (to prevent from system image flashing), and check Jetson Runtime Components and Jetson SDK Components. Then, click CONTINUE TO STEP 03

  
• Enter the IP, Username, Password which you can connect from the host to PCIe-NX150 via ssh.

  
• After several minutes, you have all the SDK components installed!

  

GPU Stress

cd /usr/src/tensorrt/bin
while true
do
  time ./trtexec --avgRuns=1000 --deploy=../data/googlenet/googlenet.prototxt --fp16 --batch=8 --iterations=1000000 --output=prob --useSpinWait --workspace=3000
done