tensorrt:https://developer.nvidia.com/tensorrt/download
cudnn:https://developer.nvidia.com/rdp/cudnn-archive
cuda:https://developer.nvidia.com/cuda-toolkit-archive
安装cuda:
直接运行可执行文件,然后取消勾选驱动(如果已经安装驱动),安装即可。
安装完毕之后添加一下环境变量:
gedit ~/.bashrc
# cuda-11.8:
export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
export CUDA_HOME=/usr/local/cuda-11.8
注意修改为自己使用的版本
接下来是安装cudnn,解压之后,执行复制操作:
sudo cp -d include/cudnn*.h /usr/local/cuda/include/
sudo cp -d lib/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
测试安装:cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
安装tensorRT:
下载包->安装->修改环境变量
export PATH=/home/cyun/Downloads/apps/TensorRT-8.5.3.1/bin:${PATH}
export LD_LIBRARY_PATH=/home/cyun/Downloads/apps/TensorRT-8.5.3.1/lib:${LD_LIBRARY_PATH}