You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: docs/docs/how-tos/use-gpus.mdx
+1-39Lines changed: 1 addition & 39 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -30,16 +30,13 @@ Overview of using GPUs on Nebari including server setup, environment setup, and
30
30
31
31
## 2. Creating environments
32
32
33
-
By default, `conda-store` will build CPU-compatible packages. To build GPU-compatible packages, we do the following.
34
33
### Build a GPU-compatible environment
35
-
By default, `conda-store` will build CPU-compatible packages. To build GPU-compatible packages, we have two options:
36
-
1.**Create the environment specification using `CONDA_OVERRIDE_CUDA` (recommended approach)**:
34
+
Create the environment specification using `CONDA_OVERRIDE_CUDA`:
37
35
38
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).
39
37
While creating a new config, click on the `**GUI <-> YAML**` Toggle to edit yaml config.
40
38
```
41
39
channels:
42
-
- pytorch
43
40
- conda-forge
44
41
dependencies:
45
42
- pytorch
@@ -57,41 +54,6 @@ At the time of writing this document, the latest CUDA version was showing as `12
57
54
Please ensure that your choice from PyTorch documentation is not greater than the highest supported version in the `nvidia-smi` output (captured above).
58
55
:::
59
56
60
-
2.**Create the environment specification based on recommendations from the PyTorch documentation**:
61
-
You can check [PyTorch documentation](https://pytorch.org/get-started/locally/) to get a quick list of the necessary CUDA-specific packages.
62
-
Select the following options to get the latest CUDA version:
63
-
- PyTorch Build = Stable
64
-
- Your OS = Linux
65
-
- Package = Conda
66
-
- Language = Python
67
-
- Compute Platform = 12.1 (Select the version that is less than or equal to the `nvidia-smi` output (see above) on your server)
The order of the channels is respected by conda, so keep pytorch at the top, then nvidia, then conda-forge.
91
-
92
-
You can use `**GUI <-> YAML**` Toggle to edit the config.
93
-
94
-
95
57
## 3. Validating the setup
96
58
You can check that your GPU server is compatible with your conda environment by opening a Jupyter Notebook, loading the environment, and running the following code:
0 commit comments