wsl2安装配置cuda10.2+cudnn8.2.1+Tensorrt8.2.0.6

组装cuda11.3

sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
sudo apt-get update

toolkit的版本一定要选择cuda版本对应的
sudo apt-get install -y cuda-toolkit-10-2

在家目录的.bashrc文件中添加以下内容, 然后执行source ~/.bashrc生效。

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

可使用nvcc -V测试，成功则显示下图信息。

加装cudnn8.2.1

cudnn官网下载10.2版本tar包，解压后分别将include及lib64下的文件复制到/usr/local/cuda对应目录下。然后命令行执行sudo ldconfig。

安装Tensorrt8.2.0.6（EA）

根据cuda版本下载对应的Tensorrt包，解压本地，在家目录的.bashrc文件中添加库文件路径, 然后执行source ~/.bashrc生效。

完事