Jetson nx 安装cuda JetPack
废话不多说,直接上流程。会输出你安装的版本号。
·
废话不多说,直接上流程
查看系统版本
head -n 1 /etc/nv_tegra_release
输出如下:
# R35 (release), REVISION: 6.0, GCID: 37391689, BOARD: t194ref, EABI: aarch64, DATE: Wed Aug 28 09:12:27 UTC 2024
表示版本系统是:L4T 35.6.0
设备类型:t194ref
添加安装源
sudo bash -c 'echo "deb https://repo.download.nvidia.com/jetson/common r35.6 main" >> /etc/apt/sources.list.d/nvidia-l4t-apt-source.list'
sudo bash -c 'echo "deb https://repo.download.nvidia.com/jetson/t194 r35.6 main" >> /etc/apt/sources.list.d/nvidia-l4t-apt-source.list'
更新
sudo apt upgrade
sudo apt update
sudo apt dist-upgrade
sudo apt install nvidia-l4t-apt-source
sudo apt update
安装
# 方式1
sudo apt install nvidia-jetpack
# 方式2:如果方式1无法安装,那么直接安装指定版本号
# 查看当前仓库中有哪些JetPack版本可用
apt-cache madison nvidia-jetpack
# 选择其中一个
sudo apt install nvidia-jetpack=<你看到的版本号>
检验安装结果
sudo apt show nvidia-jetpack | grep Version
会输出你安装的版本号
配置Cuda环境变量
vim ~/.bashrc
# 在文本末添加如下指令
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
然后更新环境变量
source ~/.bashrc
查看安装cuda是否成功:
nvcc -V
配置cuDNN
sudo cp /usr/include/cudnn* /usr/local/cuda/include
sudo cp /usr/lib/aarch64-linux-gnu/libcudnn* /usr/local/cuda/lib64
sudo chmod 777 /usr/local/cuda/include/cudnn.h
sudo chmod 777 /usr/local/cuda/lib64/libcudnn*
# 重新软链接,这里的8.6.0和8对应安装的cudnn版本号和首数字
cd /usr/local/cuda/lib64
sudo ln -sf libcudnn.so.8.6.0 libcudnn.so.8
sudo ln -sf libcudnn_ops_train.so.8.6.0 libcudnn_ops_train.so.8
sudo ln -sf libcudnn_ops_infer.so.8.6.0 libcudnn_ops_infer.so.8
sudo ln -sf libcudnn_adv_train.so.8.6.0 libcudnn_adv_train.so.8
sudo ln -sf libcudnn_adv_infer.so.8.6.0 libcudnn_adv_infer.so.8
sudo ln -sf libcudnn_cnn_train.so.8.6.0 libcudnn_cnn_train.so.8
sudo ln -sf libcudnn_cnn_infer.so.8.6.0 libcudnn_cnn_infer.so.8
sudo ldconfig
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
所有评论(0)