Update and simplify GPU environment setup instructions #602
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Removed instructions for creating a GPU-compatible environment based on PyTorch documentation as it does not show conda commands anymore. Also streamlined the section on using CONDA_OVERRIDE_CUDA and got rid of the `pytorch` channel given its deprecation: pytorch/pytorch#138506
✅ Deploy Preview for nebari-docs ready!
To edit notification comments on pull requests, go to your Netlify project configuration. |
…ari-dev/nebari-docs into update-gpu-environment-setup
marcelovilla
requested review from
Adam-D-Lewis,
dcmcand,
pavithraes and
viniciusdc
marcelovilla
commented
Oct 17, 2025
Comment on lines
-60
to
-94
| 2. **Create the environment specification based on recommendations from the PyTorch documentation**: | ||
| You can check [PyTorch documentation](https://pytorch.org/get-started/locally/) to get a quick list of the necessary CUDA-specific packages. | ||
| Select the following options to get the latest CUDA version: | ||
| - PyTorch Build = Stable | ||
| - Your OS = Linux | ||
| - Package = Conda | ||
| - Language = Python | ||
| - Compute Platform = 12.1 (Select the version that is less than or equal to the `nvidia-smi` output (see above) on your server) | ||
|
|
||
|  | ||
|
|
||
| The command `conda install` from above is: | ||
| ``` | ||
| conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia | ||
| ``` | ||
| The corresponding yaml config would be: | ||
| ``` | ||
| channels: | ||
| - pytorch | ||
| - nvidia | ||
| - conda-forge | ||
| dependencies: | ||
| - pytorch | ||
| - pytorch-cuda==12.1 | ||
| - torchvision | ||
| - torchaudio | ||
| - ipykernel | ||
| variables: {} | ||
| ``` | ||
| :::note | ||
| The order of the channels is respected by conda, so keep pytorch at the top, then nvidia, then conda-forge. | ||
|
|
||
| You can use `**GUI <-> YAML**` Toggle to edit the config. | ||
|
|
||
|
|
Member
Author
There was a problem hiding this comment.
The PyTorch docs do not show installation instructions for conda anymore:
marcelovilla
commented
Oct 17, 2025
| While creating a new config, click on the `**GUI <-> YAML**` Toggle to edit yaml config. | ||
| ``` | ||
| channels: | ||
| - pytorch |
Member
Author
There was a problem hiding this comment.
The pytorch channel has been deprecated and does not have recent versions of PyTorch: pytorch/pytorch#138506
dcmcand
approved these changes
Oct 17, 2025
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
Removed instructions for creating a GPU-compatible environment based on PyTorch documentation as it does not show conda commands anymore. Also streamlined the section on using CONDA_OVERRIDE_CUDA and got rid of the
pytorchchannel given its deprecation: pytorch/pytorch#138506Reference Issues or PRs
What does this implement/fix?
Put a
xin the boxes that applyTesting
Documentation
Access-centered content checklist
Text styling
H1or#in markdown).Non-text content
Any other comments?