|
jetson安装的jetpack 自带cuda 所以流程要比pc端简单
首先编辑环境变量 在命令行输入：
sudo vim ~/.bashrc
打开环境变量文件，然后按下键盘上的字母“a”控制vim进行插入删改操作模式（纯给新手看，当年我第一次用vim也及其蒙B）然后在文件最后插入以下环境变量代码：
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
按Esc按键退出编辑插入模式 输入wq保存并退出vim
然后刷新环境变量 方法是在命令行输入：
source ~/.bashrc
输入以下代码验证成功与否：
nvcc -V
成功则输出如下所示 具体信息视镜像版本不同而不同 我的镜像是jetpack4.6 cuda-10.2
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on sun_Feb_28_22:34:44_PST_2021
Cuda compilation tools, release 10.2, V10.2.300
Build cuda_10.2_r440.TC440_70.29663091_0
至此cuda的环境路径已经配置完毕
然后在 /var/cuda-repo-l4t-10-2-local 文件夹内可以找到cuda-samples-10-2_10.2.300-1._arm64.deb 双击安装
然后在 /usr/local/cuda-10.2/samples 文件夹中打开终端 输入以下内容以编译例程
sudo make -j6
耗时十分钟左右 完成后在 /usr/local/cuda-10.2/samples/0_Simple/vectorAddDrv 文件夹中打开终端运行例程
./vectorAddDrv
若显示Result = PASS 恭喜 你的第一个cuda例程运行成功
|