一. 查看自己的GPU版本是否支持cuda
-
打开显卡的控制面板,查看显卡是否支持cuda

二 .安装CUDA
-
去官网查看cuda版本与所需系统固件的对应关系

-
从官网下载所需的CUDA Toolkit Archive,这里下载的版本如下:

-
按默认安装下载好的安装包(只需要安装选项选择自定义,然后全选就可以)

-
安装好后,打开anaconda的terminal输入
nvcc -V,显示如下则说明安装成功

三. 安装cuDNN
-
去官网下载CUDNN,同前,注意版本号。下载地址:https://developer.nvidia.com/rdp/cudnn-download

-
下载CUDNN后,复制粘贴cuDNN里面的下面三个文件到CUDA的相应同名文件
- cudnn\cuda\bin => CUDA\v11.4\bin
- cudnn\cuda\include => CUDA\v11.4\include
- cudnn\lib\x64 => CUDA\v11.4\lib\x64

-
在系统环境变量中添加如下环境变量

四. TensorRT安装
官网安装教程: https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-zip

-
从官网下载好安装包:https://developer.nvidia.com/tensorrt.

-
解压安装包,将文件放置到自己喜欢的位置

-
添加系统环境变量

-
将TensorRT文件lib目录下的*.dll全部拷贝到CUDA安装目录的bin文件夹下

-
进入以下目录

-
进入cmd,进入自己的conda安装目录,运行以下代码
根据自己的python版本以及tensorrt版本运行以下代码
python.exe -m pip install tensorrt-8.2.3.0-cp38-none-win_amd64.whl
-
运行以下代码
python3 -m pip install C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT-8.2.3.0\graphsurgeon\graphsurgeon-0.4.6-py2.py3-none-any.whl python3 -m pip install C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT-8.2.3.0\uff\uff-0.6.9-py2.py3-none-any.whl python3 -m pip install C:\Program Files\NVIDIA GPU Computing Toolkit\TensorRT-8.2.3.0\onnx_graphsurgeon\onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl
-
测试TensorRT是否安装成功,打开以下图片目录中的项目

- 按以下操作更改项目使用的windows10 SDK至最新


- 按以下操作更改项目使用的windows10 SDK至最新
-
运行程序

-
出现以下结果说明安装成功

本文档详细介绍了如何检查GPU是否支持CUDA,以及如何一步步安装CUDA、cuDNN和TensorRT。首先,通过查看显卡控制面板确认CUDA支持。接着,从NVIDIA官网下载并安装适合系统的CUDA版本,并验证安装成功。然后,下载并安装cuDNN,将文件复制到CUDA相应目录,并设置环境变量。最后,下载TensorRT安装包,解压后配置环境变量,使用pip安装相关库,并通过测试确保TensorRT安装正确。

2万+

被折叠的 条评论
为什么被折叠?



