nvcc版本切换可执行文件,比如将cuda10.0切换为cuda11.3

首先需要确保nvcc软链接指向 /usr/local/cuda

root@I15bbb9c6ae00901c37:~# which nvcc
/usr/local/cuda/bin/nvcc

正式脚本

#!/bin/bash

# 设置CUDA版本路径
cuda_path="/usr/local/cuda"

# 获取输入的版本号
read -p "请输入要切换的CUDA版本号: " target_version

# 检查目标版本是否存在
if [ ! -d "$cuda_path-$target_version" ]; then
    echo "CUDA $target_version 不存在"
    exit 1
fi

# 切换nvcc版本
sudo rm -f $cuda_path
sudo ln -s $cuda_path-$target_version $cuda_path

# 输出切换结果
echo "已切换到CUDA $target_version"

将上述脚本保存为switch_cuda.sh文件,并赋予执行权限:

chmod +x switch_cuda.sh

然后,运行以下命令来切换到指定的CUDA版本:

sudo ./switch_cuda.sh

脚本会提示输入要切换的CUDA版本号。输入正确的版本号后,脚本将删除当前的CUDA符号链接,并创建一个指向指定版本的新符号链接。

请注意,这只会切换nvcc命令的版本,而不会切换其他CUDA组件的版本。如果需要切换其他组件(如CUDA运行时库、cuDNN等),可以手动更新相应的路径。

你可能感兴趣的:(一些有趣的应用程序,ubuntu)