Flair 0.11.3在AzureML实例导入Sentence时libcublasLt.so.11报错
解决Flair 0.11.3在AzureML计算实例中导入Sentence的CUDA依赖报错问题
问题原因
这个报错是由于PyTorch依赖的cuBLAS库与当前安装的nvidia-cublas-cu11版本不兼容导致的。cublasLtGetStatusString符号在你当前安装的11.10.3.66版本libcublasLt.so.11中不存在或版本不匹配。本地环境未安装NVIDIA相关包,使用的是CPU版PyTorch,因此没有依赖冲突;而AzureML计算实例中启用了GPU依赖,但版本适配出现问题。
解决方案
1. 卸载不兼容的NVIDIA CUDA相关包
执行以下命令清理现有冲突的依赖:
pip uninstall -y nvidia-cublas-cu11 nvidia-cuda-runtime-cu11 nvidia-cuda-nvrtc-cu11
2. 安装与PyTorch版本匹配的CUDA依赖
Flair 0.11.3通常适配PyTorch 1.12.x或1.13.x版本,对应推荐的CUDA版本为11.6。安装对应版本的依赖包:
pip install nvidia-cublas-cu11==11.9.2.110 nvidia-cuda-runtime-cu11==11.6.124 nvidia-cuda-nvrtc-cu11==11.6.124
或者直接安装适配CUDA 11.6的PyTorch完整包:
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
3. 切换到CPU版PyTorch(可选)
如果不需要GPU加速,可以直接安装CPU版PyTorch规避GPU依赖问题:
pip uninstall -y torch torchvision torchaudio pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
4. 验证修复
重新运行导入命令:
from flair.data import Sentence
内容的提问来源于stack exchange,提问作者Mohan Rayapuvari
相关产品推荐
相关产品推荐

