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.1 | CUDA 12.2 |
2023.1 | CUDA 11.8 |
2022.1 | CUDA 11.5 |
2021.1 | CUDA 11.4 |
2020.1 | CUDA 11.0 |
2019.1 | CUDA 10.2 |
2018.1 | CUDA 10.0 |
2017.1 | CUDA 9.0 |
如果没问题可以试一下:
conda install -c conda-forge pycuda==2023.1
pycuda的版本根据需要选择。我的CUDA 是11.8,所以选的是2023.1