當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博客