Update instance type to ensure newer CUDA driver version

Author: namannandan

buildspec.yml

@@ -4,7 +4,7 @@ env:
   variables:
     FRAMEWORK_VERSIONS: '2.1.0 2.2.0'
     CPU_INSTANCE_TYPE: 'ml.c4.xlarge'
-    GPU_INSTANCE_TYPE: 'ml.g4dn.12xlarge'
+    GPU_INSTANCE_TYPE: 'ml.g5.12xlarge'
     ECR_REPO: 'sagemaker-test'
     GITHUB_REPO: 'sagemaker-pytorch-serving-container'
     DLC_ACCOUNT: '763104351884'

test/integration/sagemaker/test_multi_model_endpoint_sagemaker.py

@@ -49,7 +49,7 @@ def fixture_mme_endpoint(
     sagemaker_session, image_uri, region, use_gpu, resnet18_filename, traced_resnet18_filename, s3, bucket
 ):
     try:
-        instance_type = 'ml.g4dn.xlarge' if use_gpu else 'ml.c5.xlarge'
+        instance_type = 'ml.g5.xlarge' if use_gpu else 'ml.c5.xlarge'
         endpoint_name = sagemaker.utils.unique_name_from_base("sagemaker-pytorch-serving")
 
         model_data = sagemaker_session.upload_data(