1、cuda版本应比电脑支持的最高版本低一个版本,如电脑支持11.7,应安装11.6的
使用nvidia-smi查看电脑支持的最高版本
2、由于需要执行cmake编译,因此在安装cuda时需要勾选visual studio interger(好像叫这个)。
配置参考文章:(174条消息) YOLOv5 Tensorrt Python/C++部署_yolov5 c++部署_一笑奈何LHY的博客-优快云博客
文章介绍了如何确定CUDA的合适版本,应选择略低于电脑支持最高版本的CUDA,如电脑支持11.7则安装11.6。在安装CUDA过程中需确保勾选VisualStudio集成选项,以便进行cmake编译。此外,文章还涉及YOLOv5模型使用TensorRT在Python和C++环境下的部署策略。
1、cuda版本应比电脑支持的最高版本低一个版本,如电脑支持11.7,应安装11.6的
使用nvidia-smi查看电脑支持的最高版本
2、由于需要执行cmake编译,因此在安装cuda时需要勾选visual studio interger(好像叫这个)。
配置参考文章:(174条消息) YOLOv5 Tensorrt Python/C++部署_yolov5 c++部署_一笑奈何LHY的博客-优快云博客
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
6639
2126
1900

被折叠的 条评论
为什么被折叠?