如何在anaconda虚拟环境中,安装cuda10.2 cudnn tensorrt-7.1.3.4 onnx onnxruntime(ubuntu18.04系统)

最完整的安装教程:

创建虚拟环境: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
​


{安装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







你可能感兴趣的:(软件安装,tensorflow,深度学习,ubuntu)