目前从tensorrt官网上只能找到ubuntu18.04的版本,但是我的操作系统是ubuntu20.04,实际上直接尝试直接安装ubuntu18.04的版本,这边测试是没有问题
官网下载地址:https://developer.nvidia.com/nvidia-tensorrt-download
我下载的是TensorRT 7这个版本,根据自己需要选择cuda、tensorrt对应的版本,我下载的是这个版本
https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/7.2.3/tars/TensorRT-7.2.3.4.Ubuntu-18.04.x86_64-gnu.cuda-11.1.cudnn8.1.tar.gz
注意我们下载的时候下载tar包的版本,下载完成后解压该压缩包
我本地是python3.8,所以直接安装
pip install tensorrt-7.2.3.4-cp38-none-linux_x86_64.whl
然后进行测试
python3进入python交互环境
import tensorrt as trt trt.__version__ "7.2.3.4"
打印上述版本信息说明tensorrt安装成功了。
内容版权声明:除非注明,否则皆为本站原创文章。
相关推荐
- ubuntu22.04编译安装nccl并使用nccl-tests进行单机多卡通信测试
- ubuntu22.04编译安装hwloc/libevent/ucx/openpmix/openmpi
- ubuntu安装openvpn并配置连接
- tensorflow模型占用大量显存的处理方法
- ubuntu22.04使用gpu-burn进行gpu显卡测试
- ubuntu24.04使用deepin-wine安装企业微信
- ubuntu在线安装mysql8无法使用账户密码登陆的问题
- ubuntu22.04编译安装mysql8
- ubuntu22.04编译安装nginx
- ubuntu22.04撰写定时重启和开机启动执行指定脚本
评论列表