废话不多说,直接上操作
1. 加源
安装依赖
bash
sudo apt update
sudo apt install -y software-properties-common
手动添加源
bash
sudo gedit /etc/apt/sources.list.d/nvidia-l4t-apt-source.list
然后添加相关源
bash
deb https://repo.download.nvidia.com/jetson/common r36.5 main
deb https://repo.download.nvidia.com/jetson/t234 r36.5 main
保存后执行
bash
sudo apt update
2. 安装 JetPack(完整CUDA)
执行
bash
sudo apt install -y nvidia-jetpack
3. 验证
执行
bash
ls /usr/lib/aarch64-linux-gnu/nvidia | grep cudart
应该会看到libcudart.so.12。但有些朋友会出现一些状况,发现这个动态库其实是在
bash
/usr/local/cuda-12.6/targets/aarch64-linux/lib
解决办法:
bash
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/targets/aarch64-linux/lib:$LD_LIBRARY_PATH
一劳永逸,将其放入到bashrc中
当能找到后,然后测试
bash
python3 -c "import torch; print(torch.cuda.is_available())"
这时候应该就会出现true了,说明就已经安装成功啦!!!!
可以开始你的深度学习之旅了!!!!