diff --git a/fly.toml b/fly.toml
index 1d16f66..8118fe0 100644
--- a/fly.toml
+++ b/fly.toml
@@ -18,7 +18,7 @@ primary_region = 'arn'
 
 [[vm]]
   size = 'a100-40gb'
-  memory = '1gb'
+  memory = '2gb'
   cpu_kind = 'shared'
   cpus = 1