ChatGPT-3 GPU内存不足微调flan-ul 2

2wnc66cl  于 2023-05-29  发布在  其他
关注(0)|答案(1)|浏览(191)

OutOfMemoryError:CUDA内存不足。尝试分配256.00 MiB(GPU 0; 15.78 GiB总容量; 14.99 GiB已分配; 3.50 MiB免费; PyTorch总共保留了14.99 GiB)如果保留内存>>分配内存,请尝试设置max_split_size_mb以避免碎片。有关内存管理和PYTORCH_CUDA_ALLOC_CONF,请参见文档
我有Standard_NC24s_v3单节点GPU,448 GB内存和4个GPU。但是,错误消息显示总容量为15.78GiB。微调不使用4个GPU?如何使用huggingface变压器获得Flan-UL 2微调中使用的所有4个GPU?

qyyhg6bp

qyyhg6bp1#

我通过使用以下软件包版本解决了这个问题。

!pip install transformers==4.28.1
!pip install sentencepiece==0.1.97
!pip install accelerate==0.18.0
!pip install bitsandbytes==0.37.2
!pip install torch==1.13.1

相关问题