L40 GPU facing OutOfMemoryError

#9
by Rans111 - opened

I am using 1 NVIDIA L40 GPU with 48 vRAM with vLLM Docker container. but facing OutOfMemoryError

Deploy with docker on Linux:

docker run --runtime nvidia --gpus all
--name my_vllm_container
-v ~/.cache/huggingface:/root/.cache/huggingface
--env "HUGGING_FACE_HUB_TOKEN="
-p 8000:8000
--ipc=host
vllm/vllm-openai:latest
--model unsloth/Llama-4-Scout-17B-16E-Instruct-unsloth-bnb-4bit

Error :
Torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 2.50 GiB. GPU 0 has a total capacity of 44.31 GiB of which 728.31 MiB is free. Process 2790767 has 43.59 GiB memory in use. Of the allocated memory 43.10 GiB is allocated by PyTorch, and 5.57 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation.

Your need to confirm your account before you can post a new comment.

Sign up or log in to comment