使用 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
之后重新训练不再报错
使用 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
之后重新训练不再报错