From 3d8f1779a21aaf388d1e7d60c269db1a804a3b55 Mon Sep 17 00:00:00 2001 From: Zilong Zhou Date: Tue, 17 Jun 2025 18:39:42 +0800 Subject: [PATCH] feat: use SSD with high throughput (#218) --- desktop_env/providers/aws/manager.py | 11 ++++++++++- desktop_env/providers/aws/provider.py | 9 +++++++++ 2 files changed, 19 insertions(+), 1 deletion(-) diff --git a/desktop_env/providers/aws/manager.py b/desktop_env/providers/aws/manager.py index 1502083..0b24b0f 100644 --- a/desktop_env/providers/aws/manager.py +++ b/desktop_env/providers/aws/manager.py @@ -6,7 +6,7 @@ import logging import dotenv import signal -INSTANCE_TYPE = "t3.large" +INSTANCE_TYPE = "t3.medium" # Load environment variables from .env file dotenv.load_dotenv() @@ -97,6 +97,15 @@ def _allocate_vm(region=DEFAULT_REGION): os.getenv('AWS_SECURITY_GROUP_ID') ] } + ], + "BlockDeviceMappings": [ + { + "DeviceName": "/dev/sda1", + "Ebs": { + 'VolumeType': 'io2', + 'Iops': 16000, + } + } ] } diff --git a/desktop_env/providers/aws/provider.py b/desktop_env/providers/aws/provider.py index d2a87c8..284ef2a 100644 --- a/desktop_env/providers/aws/provider.py +++ b/desktop_env/providers/aws/provider.py @@ -128,6 +128,15 @@ class AWSProvider(Provider): "DeviceIndex": 0, "Groups": security_groups } + ], + "BlockDeviceMappings": [ + { + "DeviceName": "/dev/sda1", + "Ebs": { + 'VolumeType': 'io2', + 'Iops': 16000, + } + } ] }