From c87115ee871917fae3e07b2e6e1ee39098d733aa Mon Sep 17 00:00:00 2001 From: Luiz Tauffer Date: Wed, 29 Nov 2023 14:39:03 +0100 Subject: [PATCH] Update README.md --- containers/README.md | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/containers/README.md b/containers/README.md index c85cdd0..fd33a31 100644 --- a/containers/README.md +++ b/containers/README.md @@ -105,6 +105,18 @@ If having difficulties pushing the image to ECR: The job submission must include the function arguments, which are passed and ENV vars to the running container. See [examples](https://github.com/catalystneuro/spikeinterface_cloud/tree/main/examples) of how to submit jobs with Python scripts. +If the job requires GPU, make sure to pass this extra pair of ENV variables: +``` +environment_variables.append({ + 'name': 'NVIDIA_DRIVER_CAPABILITIES', + 'value': 'all' +}) +environment_variables.append({ + 'name': 'NVIDIA_REQUIRE_CUDA', + 'value': 'cuda>=11.0' +}) +``` + # Debugging AWS Batch @@ -112,4 +124,4 @@ AWS Batch uses a series of other AWS services under the hood. If something doesn - AWS Batch > Jobs > specific-job-details - EC2 > Auto Scaling groups > specific-asg-details - EC2 > Instances > specific-instance-details > Monitoring -- CloudTrail > Event history \ No newline at end of file +- CloudTrail > Event history