在Google Colab导入aitextgen遇libtorch_cuda_cu.so缺失错误求助

这个错误是由于torchvision与当前Colab环境的CUDA版本不兼容，或相关CUDA库文件缺失导致的，按以下步骤解决：

• 先卸载现有冲突的PyTorch相关包，重新安装适配Colab默认CUDA版本的版本：

!pip uninstall -y torch torchvision torchaudio
!pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Colab当前默认CUDA版本多为11.8，上述命令会安装对应兼容的torch、torchvision版本。

• 重新安装指定版本的aitextgen及其依赖包：

!pip3 install -q aitextgen==0.5.2
!pip3 install -q torchtext==0.10.0
!pip3 install -q torchmetrics==0.6.0
!pip3 install -q pytorch-lightning==1.4.0rc0

• 重启Colab运行时（点击顶部菜单栏「Runtime」→「Restart runtime」），然后验证CUDA可用性：

import torch
print(torch.cuda.is_available())

若输出True，说明CUDA配置正常，此时再执行import aitextgen即可完成导入。

内容的提问来源于stack exchange，提问作者Peyman