a股程序化交易接口-在高版本CUDA的条件下 bitsandbytes发现低版本CUDA SETUP 完美解决方案免费用

前提条件

在安装完成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”“

文章为作者独立观点，不代表股票程序化软件自动交易接口观点