pytorch与cuda与TensorRT的版本选择

ReadyShowShow

已于 2024-07-26 16:02:40 修改

阅读量127

点赞数 2

文章标签： pytorch 人工智能 python

于 2024-07-26 15:26:04 首次发布

本文链接：https://blog.csdn.net/ReadyShowShow/article/details/140716556

版权

VScode版本

linux最新版本的vscode，可能无法进行python的调试
在这里插入图片描述
选择下载1.85 https://code.visualstudio.com/updates/v1_85

CUDA版本

在这里插入图片描述

https://developer.nvidia.com/Cuda-Toolkit-archive

由于受限于TRT的8.6（下面会说明），所以CUDA Version最高为: 12.1，而pytorch的限制只能是12.1或者是11.8。
重装cuda toolkit后，须重装pytorch

Pytorch版本

在这里插入图片描述
https://pytorch.org/get-started/locally/

选择12.1

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

TensorRT版本

在这里插入图片描述
https://developer.nvidia.com/tensorrt/download
受限于CentOS7版本，trt最高为8.6。

安装后的提示，配置路径与卸载命令：

Please make sure that
 -   PATH includes /usr/local/cuda-12.0/bin
 -   LD_LIBRARY_PATH includes /usr/local/cuda-12.0/lib64, or, add /usr/local/cuda-12.0/lib64 to /etc/ld.so.conf and run ldconfig as root

To uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-12.0/bin
To uninstall the NVIDIA Driver, run nvidia-uninstall