当NVIDIA Jetson中安装了多个CUDA时,可以通过命令,快速切换不同版本的。
这样在环境变量和代码编译时,能使用指定版本的CUDA了。
本文适用于Jetson Nano、TX1/TX2、Xavier 和 Orin系列的设备,供大家参考。
cuda参考地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/contents.html
首先看一下当前系统的CUDA 版本
根据 ls /usr/local/
的输出,系统包含以下 CUDA 版本:
cuda-12
cuda-12.2
cuda-12.6
然后查看 /usr/local/cuda
ls -l /usr/local/cuda
如果/usr/local/cuda
通过符号链接指向了 /etc/alternatives/cuda
(这是 update-alternatives
工具的默认管理路径)
这样我们可以通过系统级工具实现多版本 CUDA 切换。
步骤 1:注册所有 CUDA 版本
# 注册已安装版本(优先级数值越大优先级越高)
sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-12.6 200
sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-12.2 150
sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-12 100
步骤 2:交互式选择版本
sudo update-alternatives --config cuda
执行后会显示类似以下菜单:
There are 3 choices for the alternative cuda (providing /usr/local/cuda).Selection Path Priority Status
------------------------------------------------------------
* 0 /usr/local/cuda-12.6 200 auto mode1 /usr/local/cuda-12 100 manual mode2 /usr/local/cuda-12.2 150 manual mode3 /usr/local/cuda-12.6 200 manual modePress <enter> to keep the current choice[*], or type selection number:
输入目标版本的编号(如切换到 12.2 则输入 2
)后按回车。
步骤 3:设置环境变量
检查并修改用户配置文件( ~/.bashrc
):
# 替换为动态引用符号链接路径:
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
然后应用配置:
source ~/.bashrc
验证切换结果
nvcc -V # 检查 CUDA 编译器版本
ls -l /usr/local/cuda # 确认链接指向已更新
分享完成~
相关文章推荐:
NVIDIA Jetson 环境安装指导 PyTorch | Conda | cudnn | docker-CSDN博客
Nvidia Jetson 安装谷歌拼音googlepinyin-CSDN博客