Update DeepRVAT environment #151
Draft
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What
This PR updates the
deeprvat_env.yamlanddeeprvat_env_no_gpu.yamlprovided environment files. Specifically, updating the PyTorch and PyTorch-Lightning packs to 2.x versions.Due to the PyTorch updates, subsequent updates were required in the PyTorch Lightning
Trainerparameters, e.g. specified underpl_trainerin the respective config.yaml ,as well as, in theBaseModel.This PR also addresses Issue #16 , and both Conda and Mamba can now correctly compile cuda with the provided
deeprvat_env.yamlfile. Note: It is still recommended by the developers to use Mamba due to speed in solving the environment over Conda, e.g.mamba env create -n deeprvat -f ~/deeprvat_env.yamlTesting
Create the DeepRVAT environment from the .yaml file with various package managers and run the various DeepRVAT pipelines.