极简Tensorrt环境装配

此方法适合机型,英伟达30系显卡。

其他机型请绕步我另一篇博文

https://blog.csdn.net/qq_43515934/article/details/123897435?spm=1001.2014.3001.5501

先安pytorch,然后按tensorrt即可,在加个轮子,不用单独安cuda和cudnn
ananconda的pytorch自动装了python的cuda和cudnn,很可能有无anaconda的装法一样
tensorrt则会自动装对应的系统中的cuda,cudnn不知道装了没

1.pytorch安装

pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
 pip install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html -i http://pypi.douban.com/simple/  --trusted-host pypi.douban.com

如果速度慢的话,换手机热点即可

2.tensorrt安装

sudo dpkg -i nv-tensorrt-repo-ubuntu2004-cuda11.6-trt8.4.0.6-ea-20220212_1-1_amd64*
cd /
cd ./var
cd ./nv*
sudo apt-key add 7fa2af80.pub
# 或者
# sudo apt-key add 7fa2af80.pub
sudo apt-get update
sudo apt-get install tensorrt
python3 -m pip install numpy
sudo apt-get install python3-libnvinfer-dev
python3 -m pip install protobuf
sudo apt-get install uff-converter-tf
python3 -m pip install numpy onnx
sudo apt-get install onnx-graphsurgeon

你可能感兴趣的:(人工智能,python,shell,python,目标检测,深度学习)