c++调用python代码,使用gpu

c++调用python,使用gpu加速

1、首先要配置cuda和cudnn的环境

1、cmd窗口下nvidia-smi,查看电脑可以支持的最高cuda版本。如果nvidia-smi报错,那么需要去配置一下环境,网上有类似案例。或者通过NVIDIA控制面板/系统信息/组件里查看cuda_xxxx.dll,上面有版本号。

2、保证安装的cuda版本要小于电脑支持的版本号。我电脑最大支持cuda11,但是安装这个版本的cuda失败,安装cuda10.1成功

3、安装完cuda后,在cudnn官网,安装相应版本的cudnn

4、cmd窗口下,输入nvcc -V,若显示cuda版本不报错,则安装成功

5、在python环境下,输入python 输入import torch 输入print(torch.cuda.is_available()) 若返回true则证明环境配置成功。(ps:torch要安装gpu版本的)

6、c++调用python,使用gpu一直失败(一般是python环境配置的问题)。我实验了很多个环境,最终使用gpu加速成功。环境配置如下:

# 安装的cuda10.1,安装包适用于7.5显卡算力 参考https://zhuanlan.zhihu.com/p/544337083
certifi==2023.5.7
charset-normalizer==3.2.0
colorama==0.4.6
cycler==0.11.0
dataclasses==0.6
docopt==0.6.2
fonttools==4.38.0
future==0.18.3
idna==3.4
kiwisolver==1.4.4
matplotlib==3.5.3
numpy==1.18.5
opencv-python==4.8.0.74
packaging==23.1
pandas==1.3.5
Pillow==9.5.0

你可能感兴趣的:(c++,python,开发语言)