Skip to content

Commit

Permalink
do not land: try out new gpu worker images
Browse files Browse the repository at this point in the history
  • Loading branch information
bhearsum committed Dec 21, 2023
1 parent d7f725e commit d5ea612
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions taskcluster/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -34,28 +34,28 @@ workers:
provisioner: '{trust-domain}-{level}'
implementation: generic-worker
os: linux
worker-type: 'b-linux-v100-gpu'
worker-type: 'b-linux-v100-gpu-relops782-test'
# Use for tasks that need lots of GPU power, but not lots of disk space
# eg: translation & scoring
b-largegpu:
provisioner: '{trust-domain}-{level}'
implementation: generic-worker
os: linux
worker-type: 'b-linux-v100-gpu-4'
worker-type: 'b-linux-v100-gpu-relops782-test'
# Use for tasks that needs lots of GPU power and increased disk space
# eg: bicleaner
b-largegpu-largedisk:
provisioner: '{trust-domain}-{level}'
implementation: generic-worker
os: linux
worker-type: 'b-linux-v100-gpu-4-300gb'
worker-type: 'b-linux-v100-gpu-relops782-test'
# Use for tasks that need lots of GPU power and immensive amounts of disk space
# eg: training
b-largegpu-xlargedisk:
provisioner: '{trust-domain}-{level}'
implementation: generic-worker
os: linux
worker-type: 'b-linux-v100-gpu-4-1tb'
worker-type: 'b-linux-v100-gpu-relops782-test'
images:
provisioner: '{trust-domain}-{level}'
implementation: docker-worker
Expand All @@ -65,7 +65,7 @@ workers:
# Keys are worker type, and align with the `worker-type` entries in the
# `worker.aliases` above.
worker-configuration:
b-linux-v100-gpu:
b-linux-v100-gpu-relops782-test:
env:
GPUS: "0"
WORKSPACE: "12000"
Expand Down

0 comments on commit d5ea612

Please sign in to comment.