NVIDIA TensorRT 8.x Download | NVIDIA DeveloperCUDA:11.8
最新的pytorch2.0只支持到11.8,不建议安装最新的12.2。
cudnn:8.9.3.28
cuDNN Archive | NVIDIA Developerhttps://developer.nvidia.com/rdp/cudnn-archive
Pycuda:2022.2
pycuda · PyPIhttps://pypi.org/project/pycuda/#files
建议下载tar.gz压缩包,进入压缩包目录使用以下命令进行安装。
pip install pycuda-2022.2.2.tar.gz
Python:3.8
pycuda2022.2至少需要python3.8的版本
ptorch:2.0
或直接使用以下命令安装
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
TensorRT:8.6
选择Tensor 8.6 GA即可