VS2022配置CUDA、CuDNN和TensorRT的环境

CUDA安装

打开命令提示符,输入nvidia-smi,敲击回车。

VS2022配置CUDA、CuDNN和TensorRT的环境_第1张图片

看到自己电脑上的CUDA Version是11.7,将要下载的CUDA安装包要小于或等于11.7版本。

CUDA下载地址

我这里点击上方链接选择CUDA11.2版本

VS2022配置CUDA、CuDNN和TensorRT的环境_第2张图片

双击安装程序,路径选择默认即可。这个路径暂时存放安装中的缓存,安装结束后,这一些缓存会自动清除掉。

VS2022配置CUDA、CuDNN和TensorRT的环境_第3张图片

同意并继续

VS2022配置CUDA、CuDNN和TensorRT的环境_第4张图片

安装选项有精简安装和是自定义安装。精简安装会安装CUDA相关组件,同时也会将显卡驱动重新安装。这里不想重新安装显卡驱动,所以选择自定义安装。

VS2022配置CUDA、CuDNN和TensorRT的环境_第5张图片

选择组件的时候,将CUDA中的Nsight VSE和Visual Studio Integration取消勾选,并将Driver components里面的两个驱动组件取消,其他保持勾选。

VS2022配置CUDA、CuDNN和TensorRT的环境_第6张图片

VS2022配置CUDA、CuDNN和TensorRT的环境_第7张图片

VS2022配置CUDA、CuDNN和TensorRT的环境_第8张图片

VS2022配置CUDA、CuDNN和TensorRT的环境_第9张图片

安装成功,点关闭。

VS2022配置CUDA、CuDNN和TensorRT的环境_第10张图片

在cmd中执行nvcc -V,显示以下信息,代表安装成功。

VS2022配置CUDA、CuDNN和TensorRT的环境_第11张图片

CuDNN下载与安装

CuDNN下载地址

点击上述链接下载CuDNN,根据CUDA11.2版本,这里选择以下版本。

VS2022配置CUDA、CuDNN和TensorRT的环境_第12张图片

下载后并解压。

分别将cuda/include、cuda/lib、cuda/bin三个目录中的内容拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2对应的include、lib、bin目录下就可以了。

VS2022配置CUDA、CuDNN和TensorRT的环境_第13张图片

TensorRT的下载与安装

TensorRT下载

如果没有ivida账号,要注册一个,之后登陆。

VS2022配置CUDA、CuDNN和TensorRT的环境_第14张图片

以此点击以下两点

VS2022配置CUDA、CuDNN和TensorRT的环境_第15张图片

找到适合windows和CUDA11.2的版本,点击下载。

VS2022配置CUDA、CuDNN和TensorRT的环境_第16张图片

下载后并解压。

将 TensorRT-8.6.1.6\include中头文件 copy 到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\include

将TensorRT-8.6.1.6\lib 中所有lib文件 copy 到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\lib\x64

将TensorRT-8.6.1.6\lib 中所有dll文件copy 到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin

VS2022配置CUDA、CuDNN和TensorRT的环境_第17张图片

VS2022配置CUDA、CuDNN和TensorRT的环境_第18张图片

VS2022配置CUDA、CuDNN和TensorRT的环境_第19张图片

验证

先在CUDA的安装路径中找到deviceQuery.exe 和 bandwidthTest.exe,并复制它们的路径。

之后打开cmd,在cmd.exe中 输入cd +空格+ ‘deviceQuery.exe 所在的文件夹’,如下:

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\demo_suite

最后在这里输入deviceQuery.exe,得到如下结果说明安装成功。

VS2022配置CUDA、CuDNN和TensorRT的环境_第20张图片

同样在该路径下输入bandwidthTest.exe

VS2022配置CUDA、CuDNN和TensorRT的环境_第21张图片

以上,说明环境安装成功。

你可能感兴趣的:(深度学习,人工智能)