参考:nVidia-tensorrt-install-guide
查看环境,下载对应版本的tensorrt
需要查看的环境如下:
# 查看cuda版本
nvcc -V
# 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 查看对应的tensorflow版本
pip list|grep tensorflow
# python版本设置为3.6
我的版本如下:
本文详细介绍如何根据CUDA、cuDNN及TensorFlow版本选择并安装TensorRT。包括环境检查、TensorRT下载、安装步骤及依赖组件的配置。
参考:nVidia-tensorrt-install-guide
需要查看的环境如下:
# 查看cuda版本
nvcc -V
# 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 查看对应的tensorflow版本
pip list|grep tensorflow
# python版本设置为3.6
我的版本如下:
2万+
8263

被折叠的 条评论
为什么被折叠?
