google-deepmind / google-deepmind/weathernext

GraphCast-Lite: Running 0.25° (37 levels) FP32 model on 8GB VRAM consumer GPUs

Open
#218 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
7.7k
Forks
986
PR merge metrics
No merged PRs in 30d

Description

Hi GraphCast Team,
Thank you for your groundbreaking work!
I'd like to share a community project, GraphCast-Lite, designed to run GraphCast inference on consumer hardware by heavily optimizing memory and speed.
Key Highlights:
8GB VRAM Support: Runs the high-resolution 0.25° model (37 levels) in full FP32 precision on consumer GPUs (e.g., RTX 2080 Ti) using a Static VRAM Pool and Dual-Chunking Engine.
High Performance: Forecasts 10 days (40 steps) in ~15s on RTX 4090 (BF16) and ~40s on RTX 2080 Ti (FP16) via TensorRT.
Lightweight: Replaces heavy dependencies with a pure PyTorch/TensorRT inference pipeline.
Repo: [](https://github.com/VectorElectron/graphcast-lite)
Hope this helps more researchers experiment with GraphCast without needing enterprise GPUs!
Best regards,

Contributor guide

Open the contributing guide

Research direction

Review the linked VectorElectron/graphcast-lite repository and the issue's claims about PyTorch, TensorRT, VRAM usage, and forecast speed. The issue names no file, test, or requested change; completion would require clarifying whether the project should be integrated, documented, or merely acknowledged.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning, performance
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Quiet
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.