Use the local Survival mode. The opening asks you to keep an eye on heat as well as oxygen. Establish a small underwater base with a place to store materials, then use it to make longer trips back to the surface or deeper below it.
