使用 transformers  Trainer 进行模型训练的时候,报了如下错误:

Could not load library libcudnn_cnn_train.so.8. Error: /usr/local/cuda-12.1/lib64/libcudnn_cnn_train.so.8

解决方法:

把如下这个环境变量删除

执行如下命令进行删除:

unset LD_LIBRARY_PATH

之后重新训练不再报错

Logo

一站式 AI 云服务平台

更多推荐