前提条件
在安装完成kohya_ss项目后，准备做大模型训练的过程中遇到以下错误
/root/anaconda3/envs/kohyass/lib/python10/site-packages/bitsandbytes/cuda_setup/paths.py:93:UserWarning:/root/anaconda3/envs/kohyassdidnotcontainlibcudart.soasexpected!Searchingfurtherpaths…warn}warn.Youwillbeonlytouse8-bitoptimizersandquantizationroutines!!CUDASETUP:DetectedCUDAversion102CUDASETUP:TODO:compilelibraryforspecificversion:libbitsandbytes_cuda10soCUDASETUP:Defaultingtolibbitsandbytes.so…CUDASETUP:CUDAdetectionfailed.EitherCUDAdrivernotinstalled,CUDAnotinstalled,oryouhavemultipleconflictingCUDAlibraries!CUDASETUP:Ifyoucompiledfromsource,tryagainwithmakeCUDA_VERSION=DETECTED_CUDA_VERSIONforexample,makeCUDA_VERSION=11
解决方案如下：pip卸载bitsandbytes
pip安装bitsandbytes-cuda116--安装你cuda对应版本的bitsandbytes
我安装的为“pipinstallbitsandbytes-cuda113==0.20.post2”“
文章为作者独立观点，不代表股票交易接口观点
股民评论