tensorRT安装

tensorRT安装
官网链接: tensorRT下载

1、根据需要选择对应的版本
根据需要选择对应的版本
2、根据系统和CUDA版本选择选一个
在这里插入图片描述
3、将文件解压
在这里插入图片描述
4、将lib目录和include目录中的文件内容都拷贝到cuda对应的目录文件下

例如我的是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

5、在python文件夹中有对应的Python安装库
在这里插入图片描述
6、在samples中有相关的使用示例
在这里插入图片描述
我下的版本是TensorRT-10.9.0.34.Windows.win10.cuda-11.8
百度网盘下载:TensorRT-10.9.0.34.Windows.win10.cuda-11.8.zip
提取码: kf61

pycuda安转的那些事
使用 pip install pycuda
报错:
ERROR: Failed to build installable wheels for some pyproject.toml based projects (pycuda)
有可能是pycuda的版本与电脑状的CUDA不对应

PyCUDA 版本CUDA 版本
2024.1CUDA 12.2
2023.1CUDA 11.8
2022.1CUDA 11.5
2021.1CUDA 11.4
2020.1CUDA 11.0
2019.1CUDA 10.2
2018.1CUDA 10.0
2017.1CUDA 9.0

如果没问题可以试一下:

conda install -c conda-forge pycuda==2023.1

pycuda的版本根据需要选择。我的CUDA 是11.8,所以选的是2023.1

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值