Skip to content

Commit c346e43

Browse files
Update GPU instructions (#32959)
1 parent bb759e6 commit c346e43

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

docs/machine-learning/how-to-guides/install-gpu-model-builder.md

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
---
22
title: How to install GPU support in Model Builder
33
description: Learn how to install GPU support in Model Builder
4-
ms.date: 11/10/2022
4+
ms.date: 12/08/2022
55
author: luisquintanilla
66
ms.author: luquinta
77
ms.topic: how-to
@@ -20,17 +20,15 @@ Learn how to install the GPU drivers to use your GPU with Model Builder.
2020
## Prerequisites
2121

2222
- [Model Builder Visual Studio extension](install-model-builder.md). The extension is built into Visual Studio as of version 16.6.1.
23-
- NVIDIA developer account. If you don't have one, [create a free account](https://developer.nvidia.com/developer-program).
2423
- Make sure the appropriate [driver](https://www.nvidia.com/drivers) is installed for the GPU.
2524

26-
**Image classification only**
25+
### Image classification only
2726

2827
- [Model Builder Visual Studio GPU support extension](https://marketplace.visualstudio.com/items?itemName=MLNET.ModelBuilderGPU2022)
29-
30-
## Install dependencies
31-
32-
1. Install [CUDA v10.1](https://developer.nvidia.com/cuda-10.1-download-archive-update2). Make sure you install CUDA v10.1, not any other newer version.
33-
1. Install [cuDNN v7.6.4 for CUDA 10.1](https://developer.nvidia.com/rdp/cudnn-download). You cannot have multiple versions of cuDNN installed. After downloading cuDNN v7.6.4 zip file and unpacking it, copy `<CUDNN_zip_files_path>\cuda\bin\cudnn64_7.dll` to `<YOUR_DRIVE>\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin`.
28+
- NVIDIA developer account. If you don't have one, [create a free account](https://developer.nvidia.com/developer-program).
29+
- Install dependencies
30+
- Install [CUDA v10.1](https://developer.nvidia.com/cuda-10.1-download-archive-update2). Make sure you install CUDA v10.1, not any other newer version.
31+
- Install [cuDNN v7.6.4 for CUDA 10.1](https://developer.nvidia.com/rdp/cudnn-download). You cannot have multiple versions of cuDNN installed. After downloading cuDNN v7.6.4 zip file and unpacking it, copy `<CUDNN_zip_files_path>\cuda\bin\cudnn64_7.dll` to `<YOUR_DRIVE>\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin`.
3432

3533
## Troubleshooting
3634

0 commit comments

Comments
 (0)