Not Rendering with Nvidia GPU

CPU AMD 5950x GPU Nvidia RTX3090 with Driver 535-server System Ubuntu 20.04 Python 3.10 I am running python -m pysc2.bin.agent --map Simple64 with pySC2 4.0.0 with SC2.4.10 and cannot render the scene in pygame. The screenshot is as follows.

One suggestions is to use pygame==1.9.6 but there's no wheel of it from python3.8 to 3.10 and I got several issues related to C++ dependencies, so I failed to install it and use pygame==2.5.2.