Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Base Learner Teams #298

Open
kyle-messier opened this issue Mar 4, 2024 · 2 comments
Open

Base Learner Teams #298

kyle-messier opened this issue Mar 4, 2024 · 2 comments
Labels
development models Related to base or meta learner fitting

Comments

@kyle-messier
Copy link
Collaborator

No description provided.

@kyle-messier kyle-messier converted this from a draft issue Mar 4, 2024
@kyle-messier
Copy link
Collaborator Author

  1. xgboost: @eva0marques @Sanisha003
  2. [C]NN: @sigmafelix @sigmafelix
  3. prestoGP: @dzilber @dawranadeep

@sigmafelix
Copy link
Collaborator

sigmafelix commented Jun 10, 2024

GPU settings

  • Assumption: we are in a HPC interactive session
  1. XGBoost
  • Setting up a user library path where you want (in my case, ~/r-libs)
  • Download prebuilt GPU version library at https://github.com/dmlc/xgboost/releases/tag/v2.0.3
  • Run command R CMD INSTALL [downloaded tarball file path] --library=~/r-libs
  • Bind CUDA (please refer to the post on Teams)
  1. Brulee
  • Bind CUDA
  • Install PyTorch (*note -- R torch currently supports CUDA 11.8 is supported on HPC)
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
  • In R, run pak::pak("tidymodels/brulee@gpu")

@mitchellmanware mitchellmanware added models Related to base or meta learner fitting development labels Jan 27, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
development models Related to base or meta learner fitting
Projects
None yet
Development

No branches or pull requests

3 participants