You can now use the geospatial image within SageMaker Studio’s notebook with GPU-based instances. Support for GPU-based instances with the geospatial image makes it easier for data scientists and machine learning (ML) engineers to build, train, and deploy ML models using geospatial data.
Source:: Amazon AWS