Skip to content

Commit 36d9a59

Browse files
committed
Improve writing
1 parent 187031f commit 36d9a59

File tree

1 file changed

+13
-19
lines changed

1 file changed

+13
-19
lines changed

docs/docs/how-tos/use-gpus.mdx

Lines changed: 13 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -31,27 +31,21 @@ Overview of using GPUs on Nebari including server setup, environment setup, and
3131
## 2. Creating environments
3232

3333
### Build a GPU-compatible environment
34-
Create the environment specification using `CONDA_OVERRIDE_CUDA`:
35-
36-
Conda-store provides an alternate mechanism to enable GPU environments via the setting of an environment variable as explained in the [conda-store docs](https://conda.store/conda-store-ui/tutorials/create-envs#set-environment-variables).
37-
While creating a new config, click on the `**GUI <-> YAML**` Toggle to edit yaml config.
38-
```
39-
channels:
40-
- conda-forge
41-
dependencies:
42-
- pytorch
43-
- ipykernel
44-
variables:
45-
CONDA_OVERRIDE_CUDA: "12.1"
46-
```
47-
Alternatively, you can configure the same config using the UI.
48-
49-
Add the `CONDA_OVERRIDE_CUDA` override to the variables section to tell conda-store to build a GPU-compatible environment.
34+
conda-store provides an alternate mechanism to enable GPU environments via the setting of an environment variable as explained in the [conda-store docs](https://conda.store/conda-store-ui/tutorials/create-envs#set-environment-variables).
35+
Create the environment specification using `CONDA_OVERRIDE_CUDA` by creating a new environment and clicking on the `**GUI <-> YAML**` toggle to edit the yaml config.
36+
```yaml
37+
channels:
38+
- conda-forge
39+
dependencies:
40+
- pytorch
41+
- ipykernel
42+
variables:
43+
CONDA_OVERRIDE_CUDA: "12.4"
44+
```
45+
Alternatively, you can configure the same variable using the UI.
5046
5147
:::note
52-
At the time of writing this document, the latest CUDA version was showing as `12.1`. Please follow the steps below to determine the latest override value for the `CONDA_OVERRIDE_CUDA` environment variable.
53-
54-
Please ensure that your choice from PyTorch documentation is not greater than the highest supported version in the `nvidia-smi` output (captured above).
48+
At the time of writing this document, the latest CUDA version was showing as `12.4`. Please follow the steps [above](understanding-gpu-setup-on-the-server) to determine the highest supported version to use as an override value for the `CONDA_OVERRIDE_CUDA` environment variable.
5549
:::
5650

5751
## 3. Validating the setup

0 commit comments

Comments
 (0)