Skip to content

Commit ca08ef4

Browse files
author
betterpig
committed
add vs2019 for cuda >= 11.2 and method of specify cuda version
1 parent b72b7dd commit ca08ef4

File tree

1 file changed

+12
-7
lines changed

1 file changed

+12
-7
lines changed

docs/install/compile/windows-compile.md

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,30 +8,30 @@
88

99
* **Windows 7/8/10 专业版/企业版 (64bit)**
1010
* **Python 版本 3.6/3.7/3.8/3.9 (64 bit)**
11-
* **Visual Studio 2017 社区版/专业版/企业版**
11+
* **Visual Studio 2017/2019 社区版/专业版/企业版**
1212

1313
## 选择 CPU/GPU
1414

1515
* 如果你的计算机硬件没有 NVIDIA® GPU,请编译 CPU 版本的 PaddlePaddle
1616

17-
* 如果你的计算机硬件有 NVIDIA® GPU,推荐编译 GPU 版本的 PaddlePaddle,建议安装 **CUDA 10.1/10.2/11.0/11.1/11.2**
17+
* 如果你的计算机硬件有 NVIDIA® GPU,推荐编译 GPU 版本的 PaddlePaddle,建议安装 **CUDA 10.1/10.2/11.0/11.1/11.2/11.6**
1818

1919
## 本机编译过程
2020

21-
1. 安装必要的工具 cmake, git, python, Visual studio 2017:
21+
1. 安装必要的工具 cmake, git, python, Visual studio 2017/2019
2222

2323
> **cmake**:建议安装 CMake3.17 版本, 官网下载[链接](https://cmake.org/files/v3.17/cmake-3.17.0-win64-x64.msi)。安装时注意勾选 `Add CMake to the system PATH for all users`,将 CMake 添加到环境变量中。
2424
2525
> **git**:官网下载[链接](https://github.com/git-for-windows/git/releases/download/v2.35.1.windows.2/Git-2.35.1.2-64-bit.exe),使用默认选项安装。
2626
2727
> **python**:官网[链接](https://www.python.org/downloads/windows/),可选择 3.6/3.7/3.8/3.9 中任一版本的 Windows installer(64-bit)安装。安装时注意勾选 `Add Python 3.x to PATH`,将 Python 添加到环境变量中。
2828
29-
> **Visual studio 2017**:官网[链接](https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/#visual-studio-2017-and-other-products),需要登录后下载,建议下载 Community 社区版。在安装时需要在工作负荷一栏中勾选 `使用 C++的桌面开发``通用 Windows 平台开发`,并在语言包一栏中选择 `英语`
29+
> **Visual studio**需根据 CUDA 版本选择对应的 Visual studio 版本,当只编译 CPU 版本或者 CUDA 版本 < 11.2 时,安装 VS2017;当 CUDA 版本 >= 11.2 时,安装 VS2019。官网[链接](https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/),需要登录后下载,建议下载 Community 社区版。在安装时需要在工作负荷一栏中勾选 `使用 C++的桌面开发``通用 Windows 平台开发`,并在语言包一栏中选择 `英语`
3030
31-
2. 在 Windows 桌面下方的搜索栏中搜索 `x64 Native Tools Command Prompt for VS 2017``适用于 VS 2017 的 x64 本机工具命令提示符`,右键以管理员身份打开终端。之后的命令均在该终端中执行。
31+
2. 在 Windows 桌面下方的搜索栏中搜索 `x64 Native Tools Command Prompt for VS 2017``适用于 VS 2017 的 x64 本机工具命令提示符`(若为 VS2019 版本则将 2017 改为 2019),右键以管理员身份打开终端。之后的命令均在该终端中执行。
3232

3333
3. 使用`pip`命令安装 Python 依赖:
34-
* 通过 `python --version` 检查默认 python 版本是否是预期版本,因为你的计算机可能安装有多个 python,你可通过修改系统环境变量的顺序来修改默认 Python 版本。
34+
* 通过 `python --version` 检查默认 python 版本是否是预期版本,因为你的计算机可能安装有多个 python,可通过修改系统环境变量的顺序来修改默认 Python 版本。
3535
* 安装 numpy, protobuf, wheel, ninja
3636
```
3737
pip install numpy protobuf wheel ninja
@@ -72,7 +72,12 @@
7272
其他编译选项含义请参见[编译选项表](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/install/Tables.html#Compile)。
7373
7474
> 注意:
75-
> 1. 如果本机安装了多个 CUDA,将使用最新安装的 CUDA 版本,且无法指定。
75+
> 1. 如果本机安装了多个 CUDA,将使用最新安装的 CUDA 版本。若需要指定 CUDA 版本,则需要设置环境变量和 cmake 选项,例如:
76+
```
77+
set CUDA_TOOLKIT_ROOT_DIR=C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2
78+
set PATH=%CUDA_TOOLKIT_ROOT_DIR:/=\%\bin;%CUDA_TOOLKIT_ROOT_DIR:/=\%\libnvvp;%PATH%
79+
cmake .. -GNinja -DWITH_GPU=ON -DCUDA_TOOLKIT_ROOT_DIR="%CUDA_TOOLKIT_ROOT_DIR%"
80+
```
7681
> 2. 如果本机安装了多个 Python,将使用最新安装的 Python 版本。若需要指定 Python 版本,则需要指定 Python 路径,例如:
7782
```
7883
cmake .. -GNinja -DWITH_GPU=ON -DPYTHON_EXECUTABLE=C:\Python38\python.exe -DPYTHON_INCLUDE_DIR=C:\Python38\include -DPYTHON_LIBRARY=C:\Python38\libs\python38.lib

0 commit comments

Comments
 (0)