Skip to content

Conversation

@cbcase
Copy link

@cbcase cbcase commented Aug 19, 2024

What is says on the tin. Only things that aren't copy-paste:

  • NV base image name has changed slightly (cudnn version no longer present)
  • URL for pytorch wheels has to tweak slightly
  • No longer installing apex in the environments base image, since we will re-install it (from our pypi server) in the augment image

@cbcase cbcase requested review from marcmac and mmonaco August 19, 2024 23:30
@cbcase cbcase merged commit f7ea8ed into augment_image Aug 19, 2024
@cbcase cbcase deleted the augment-image-cuda124-pytorch240 branch August 19, 2024 23:34
@cbcase
Copy link
Author

cbcase commented Aug 19, 2024

FYI that this image is now pushed and available at au-docker-reg.tenant-augment-eng.ord1.ingress.coreweave.cloud/dai_environments:ubuntu22.04-cuda-12.4-py-3.9-pytorch-2.4.0-gpt-neox-deepspeed-gpu-0.27.1

@mmonaco
Copy link

mmonaco commented Aug 20, 2024

Are you going to run some tests before we update the augment_gpu images?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants