在Visual Studio中进行Python,C++和CUDA混合编程：Windows操作系统下的环境配置

fmy_xfk

于 2024-08-13 16:38:23 发布

阅读量633

点赞数 6

本文链接：https://blog.csdn.net/fmy_xfk/article/details/141165207

版权

在Visual Studio和PyCharm中进行Python,C++和CUDA混合编程：Windows操作系统下的环境配置

项目需求

现有一Python开发的项目，部分代码需要改用C++运行提速。C++部分中还有一些模块需要改用CUDA运行以实现并行化。整个项目只需运行于Windows操作系统。

整体方案

进行CUDA开发，生成链接库文件(.lib和.dll)；
编制CUDA部分的测试单元，验证其正确性；
利用pybind11构建C++编写的Python库，并在其中引入CUDA部分的库文件；
编写Python程序，验证整个项目的正确性。

Visual Studio和CUDA的安装

安装Visual Studio，建议2019或者2022版本；
安装CUDA，版本根据实际需求来，建议11.6及以上；

注意：必须先安装Visual Studio，再安装CUDA；安装CUDA时，务必勾选Visual Studio Integration项目。

构建CUDA程序链接库

在Visual Studio中新建工程，选择CUDA 11.6 Runtime；
在“项目”—“属性”—“常规”页面，将“配置类型”改为“动态库(.dll)”；
在“项目”—“生成自定义…”页面，勾选“CUDA 11.6(.targets,.props)”
此后便可以编制代码（.cu源文件和.cuh头文件），注意头文件中需要对要导出的函数添加形如extern "C" __declspec(dllexport) void functionName(...);的内容，否则后面的步骤会出错；
生成项目，此时需要找到并记录生成文件的位置，例如"…/x64/Debug/xxx.lib"。

注意：这里11.6是笔者安装的版本，请根据实际版本操作；

构建CUDA测试单元

在Visual Studio中，在同一解决方案下新建工程，选择C++空项目；
创建文件，通过下列代码导入头文件和库文件；

#include "头文件.cuh"  //注意这里添加头文件,用的相对路径
#pragma comment (lib, "库文件路径.lib")

构建基于pybind11的C++编写的Python库

首先，利用pip安装pybind11并克隆源码：

pip install pybind11
git clone https://github.com/pybind/pybind11.git

这里省略了虚拟环境的相关内容，有需要者自行创建；

在刚刚执行命令的文件夹下，创建CMakeLists.txt：

cmake_minimum_required(VERSION 3.2) 
project(工程名)
add_subdirectory(pybind11)
pybind11_add_module(工程名 源码文件.cpp)

将“工程名”和“源码文件.cpp”替换成实际名称；

使用cmake构建Visual Studio工程，对于VS2019版本，将G参数换成"Visual Studio 16 2019"：

cmake . -G "Visual Studio 17 2022" -A x64

执行完毕后，即可在文件夹中看到Visual Studio的工程文件，包含：工程名.sln，工程名.vcxproj，ALL_BUILD.vcxproj和ZERO_CHECK.vcxproj。我们只需要工程名.vcxproj，把它添加到原有的那个解决方案（不是这一步生成的解决方案）中即可。