cudart link错误自动修复脚本

问题

在conda安装cuda环境时

mamba install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y
mamba install nvidia/label/cuda-12.1.0::cuda

安装的envs/xxx/lib/libcudart.so 会link到错的

如果手动修复太复杂，我写了一个脚本自动修复

脚本

#!/bin/bash# 检查是否提供了环境名称作为参数
if [ "$#" -ne 1 ]; thenecho "Usage: $0 <environment_name>"exit 1
fienv_name=$1
env_path="/mnt/data/wangziyi/miniconda3/envs/$env_name/lib"# 检查环境路径是否存在
if [ ! -d "$env_path" ]; thenecho "Error: The environment path $env_path does not exist."exit 1
fi# 使用find命令和正则表达式来查找最长的libcudart.so文件名
cudart_so_file=$(find "$env_path" -type f -regextype posix-extended -regex "$env_path/libcudart\.so\.[0-9]+\.[0-9]+\.[0-9]+" -printf '%f\n' | sort -rV | head -n 1)# 检查是否找到了文件
if [ -z "$cudart_so_file" ]; thenecho "Error: No file matching libcudart.so.<version> was found in the environment."exit 1
fi# 创建或更新libcudart.so的符号链接
ln -sf "$env_path/$cudart_so_file" "$env_path/libcudart.so"echo "Symbol link for libcudart.so has been updated to $cudart_so_file."