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等),可以手动更新相应的路径。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值