如何在anaconda虚拟环境中，安装cuda10.2 cudnn tensorrt

创建虚拟环境 conda create -n xxx python 3.6

激活虚拟环境 source activate xxx

安装cuda10.2: conda install cudatoolkit 10.2

安装cudnn :conda install cudnn(可以不加版本 自动匹配cuda)

安装pytorch:

conda install pytorch 1.7.1 torchvision 0.8.2 torchaudio 0.7.2 cudatoolkit 10.2 -c pytorch

安装自己合适的版本链接 Previous PyTorch Versions | PyTorch
 

{安装tensorrt 首先需要下载tensorrt的包 如 TensorRT-7.1.3.4.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn8.0.tar.gz
在home路径下解压得到TensorRT-7.1.3.4的文件夹 将里边的lib绝对路径添加到环境变量中
gedit ~/.bashrc
添加环境变量
export LD_LIBRARY_PATH $LD_LIBRARY_PATH:/home/zhangsan/TensorRT-7.1.3.4/lib:$LD_LIBRARY_PATH
生效
source ~/.bashrc
安装包的下载以及解压 不是在虚拟环境中操作的 是在用户目录下操作的}

配置好环境变量之后 激活anacoda虚拟环境
安装tensorrt
cd TensorRT-7.1.3.4/python/
pip install tensorrt-7.1.3.4-cp36-none-linux_x86_64.whl cp36 根据python版本解释器选择
安装UFF 与tensorflow相关 可不装 若用pytorh框架
cd TensorRT-7.1.3.4/uff
pip install uff-0.6.9-py2.py3-none-any.whl

安装graphsurgeon
cd TensorRT-7.1.3.4/graphsurgeon
pip install graphsurgeon-0.4.5-py2.py3-none-any.whl

安装onnx与onnxruntime:
conda install -c conda-forge onnx
conda install -c conda-forge onnxruntime

也可用Pip的安装方式
pip install onnx
pip install onnxruntime

最完整的安装 
创建虚拟环境 conda create -n xxx python 3.6
激活虚拟环境 source activate xxx
安装cuda10.2: conda install cudatoolkit 10.2
安装cudnn :conda install cudnn(可以不加版本 自动匹配cuda)
安葬pytorch ​
安装pytorch:
conda install pytorch 1.7.1 torchvision 0.8.2 torchaudio 0.7.2 cudatoolkit 10.2 -c pytorch
[点击并拖拽以移动]
安装自己合适的版本链接 Previous PyTorch Versions | PyTorch