diff --git a/.drone.yml b/.drone.yml
index 363166e..52eef53 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -9,6 +9,7 @@ steps:
   - name: install
     image: pytorch/pytorch:latest
     privileged: true
+    shm_size: 1073741824
     environment:
       mount_command:
         from_secret: mount_command