Sign inSign up

sinanozel/kubyterlab-llm

By sinanozel

•Updated 8 months ago

A custom JupyterLab container for AI-related work, focusing on using GPU memory.

Image
Machine learning & AI
Developer tools
Data science
0

5.2K

sinanozel/kubyterlab-llm repository overview

This is a custom JupyterLab specifically for working with GPUs, tailored towards LLM, NLP and OCR applications. See here for more info: https://github.com/sinan-ozel/jupyterlab-on-kubernetes/tree/main/kubyterlab-llm⁠

With docker: docker run --gpus all --rm -it sinanozel/kubyterlab-llm:25.09 python -c "import torch; print('CUDA available:', torch.cuda.is_available());"

If you see True, you can use this JupyterLab container with CUDA, meaning that you can use the host computer's GPU memory for AI work.

The host machine needs to have CUDA and nvidia-smi and nvidia-container-toolkit installed.

With Kubernetes, start a cluster with some nodes with GPUs and node images that have CUDA and nvidia-smi and nvidia-container-toolkit installed. Consider using the provision configuration from my IaC template repo, https://github.com/sinan-ozel/iac⁠

Tag summary

Content type

Image

Digest

sha256:42c68c8e8…

Size

25.7 GB

Last updated

8 months ago

docker pull sinanozel/kubyterlab-llm:26.01