Skip to content

Commit 9d4baee

Browse files
jarmak-nvbdice
andauthored
Add CUDA 12.2 in nightlies (#483)
Co-authored-by: Bradley Dice <[email protected]>
1 parent 61b1c6e commit 9d4baee

File tree

2 files changed

+11
-6
lines changed

2 files changed

+11
-6
lines changed

_includes/selector.html

+7-4
Original file line numberDiff line numberDiff line change
@@ -370,8 +370,8 @@
370370

371371
// all possible values
372372
python_vers: ["3.9", "3.10"],
373-
cuda_vers: ["11.2", "11.8", "12.0"],
374-
pip_cuda_vers: ["11.2 - 11.8", "12.0"],
373+
cuda_vers: ["11.2", "11.8", "12.0", "12.2"],
374+
pip_cuda_vers: ["11.2 - 11.8", "12"],
375375
methods: ["Conda", "pip", "Docker"],
376376
releases: ["Stable", "Nightly"],
377377
img_loc: ["NGC", "Dockerhub"],
@@ -627,9 +627,9 @@
627627
},
628628
getpipNotes() {
629629
var notes = [];
630-
var install_location_notes = "<code>cuDF</code>, <code>dask-cuDF</code>, <code>cuML</code>, <code>cuGraph</code>, <code>cuSpatial</code>, <code>cuProj</code>, <code>cuxfilter</code>, <code>cuCIM</code>, and <code>RAFT</code> pip packages are hosted by NVIDIA<br>"
630+
var install_location_notes = "RAPIDS pip packages are hosted by NVIDIA<br>"
631631
notes = [...notes, install_location_notes,
632-
'pip installation supports Python <code>3.9</code> and <code>3.10</code><br>'];
632+
'pip installation supports only Python <code>3.9</code> and <code>3.10</code><br>'];
633633

634634
return notes.map(note => this.note_prefix + " " + note);
635635
},
@@ -668,6 +668,7 @@
668668
var isDisabled = false;
669669
if (this.active_additional_packages.includes("TensorFlow") && (cuda_version !== "12.0")) isDisabled = true;
670670
if (this.active_method === "Docker" && cuda_version < "11.8") isDisabled = true;
671+
if (this.active_release != "Nightly" && cuda_version === "12.2") isDisabled = true;
671672
return isDisabled;
672673
},
673674
disableUnsupportedPython(python_version) {
@@ -696,6 +697,8 @@
696697
},
697698
releaseClickHandler(e, release) {
698699
if (this.isDisabled(e.target)) return;
700+
// Remove below after we support 12.2 in stable
701+
if (this.active_cuda_ver === "12.2" && release === "Stable") this.active_cuda_ver = "12.0";
699702
this.active_release = release;
700703
},
701704
imgTypeClickHandler(e, type) {

install/install.md

+4-2
Original file line numberDiff line numberDiff line change
@@ -137,14 +137,15 @@ All provisioned systems need to be RAPIDS capable. Here's what is required:
137137
- <i class="fas fa-check-circle"></i> [CUDA 11.5](https://developer.nvidia.com/cuda-11-5-0-download-archive){: target="_blank"} with Driver 495.29.05 or newer
138138
- <i class="fas fa-check-circle"></i> [CUDA 11.8](https://developer.nvidia.com/cuda-11-8-0-download-archive){: target="_blank"} with Driver 520.61.05 or newer
139139
- <i class="fas fa-check-circle"></i> [CUDA 12.0](https://developer.nvidia.com/cuda-12-0-1-download-archive){: target="_blank"} with Driver 525.60.13 or newer **see CUDA 12 section below for notes on usage**
140+
- <i class="fas fa-check-circle"></i> [CUDA 12.2](https://developer.nvidia.com/cuda-12-2-2-download-archive){: target="_blank"} with Driver 535.86.10 or newer **see CUDA 12 section below for notes on usage**
140141

141142
**Note**: RAPIDS is tested with and officially supports the versions listed above. Newer CUDA and driver versions may also work with RAPIDS. See [CUDA compatibility](https://docs.nvidia.com/deploy/cuda-compatibility/index.html) for details.
142143

143144
## **CUDA 12 Support**
144145

145146
### **Docker and Conda**
146147

147-
- <i class="fas fa-info-circle"></i> CUDA 12 conda packages and Docker images currently support CUDA 12.0
148+
- <i class="fas fa-info-circle"></i> Stable CUDA 12 conda packages and Docker images currently support CUDA 12.0. Additionally, nightly versions now support CUDA 12.2
148149
- <i class="fas fa-info-circle"></i> CUDA 11 conda packages and Docker images can be used on a system with a CUDA 12 driver because they include their own CUDA toolkit
149150

150151
### **pip**
@@ -334,8 +335,9 @@ Windows users can now tap into GPU accelerated data science on their local machi
334335
1. Install WSL2 and the Ubuntu 22.04 package [using Microsoft's instructions](https://docs.microsoft.com/en-us/windows/wsl/install){: target="_blank"}.
335336
2. Install the [latest NVIDIA Drivers](https://www.nvidia.com/download/index.aspx){: target="_blank"} on the Windows host.
336337
3. Log in to the WSL2 Linux instance.
337-
4. Follow [this helpful developer guide](https://docs.nvidia.com/cuda/wsl-user-guide/index.html#cuda-support-for-wsl2){: target="_blank"} and then install the WSL-specific [CUDA 11.8](https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_local){: target="_blank"} or [CUDA 12.0](https://developer.nvidia.com/cuda-12-0-1-download-archive?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_local){: target="_blank"} Toolkit without drivers into the WSL2 instance.
338+
4. Follow [this helpful developer guide](https://docs.nvidia.com/cuda/wsl-user-guide/index.html#cuda-support-for-wsl2){: target="_blank"} and then install the WSL-specific [CUDA 11](https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_local){: target="_blank"} or [CUDA 12](https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_local){: target="_blank"} Toolkit without drivers into the WSL2 instance.
338339
- The installed CUDA Toolkit version must match the pip wheel version (`-cu11` or `-cu12`)
340+
- Any CUDA 12 CTK will work with RAPIDS `-cu12` pip packages
339341
5. Install RAPIDS pip packages on the WSL2 Linux Instance using the [release selector](#selector) commands.
340342
6. Run this code to check that the RAPIDS installation is working:
341343
```

0 commit comments

Comments
 (0)