Skip to content

Conversation

@marcmac
Copy link

@marcmac marcmac commented Oct 12, 2023

I don't intend to merg to main, but this is easiest to look at. One bit I'm not sure about, the DEEPSPEED_VERSION

@marcmac marcmac requested a review from cbcase October 12, 2023 18:03
@marcmac marcmac marked this pull request as draft October 12, 2023 18:03
Makefile Outdated
export GPU_DEEPSPEED_ENVIRONMENT_NAME := $(CUDA_113_PREFIX)pytorch-1.10-deepspeed-$(DEEPSPEED_VERSION)$(GPU_SUFFIX)
export GPU_GPT_NEOX_DEEPSPEED_ENVIRONMENT_NAME := $(CUDA_113_PREFIX)pytorch-1.10-gpt-neox-deepspeed$(GPU_SUFFIX)
export TORCH_PIP_DEEPSPEED_GPU := torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio==0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
DEEPSPEED_VERSION := 0.7.0
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@cbcase I'm thinking I should accept the upstream change to 0.8.3, even though we re-install it.

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.

2 participants