当前位置:   article > 正文

2024年YOLOv8安装配置+在个人工作目录下使用YOLOv8_yolov8安装gpu版本

yolov8安装gpu版本

环境

硬件环境:实验室集群-显卡RTX4090

容器操作系统:Ubuntu18.04

软件:Anaconda下py3.9+pytorch2.1.2(cuda11.8版本)

Tips:如果显卡是RTX4090且安装的驱动版本最高支持cuda11.8及以上,那么安装pytorch-gpu版本的时候需要选择安装其中的cuda11.8及以上的版本,低版本的可能会有问题,另外不需要事先在集群环境中安装cuda和cudnn,pytorch-gpu版本自带cuda工具包。安装pytorch可参考该博客:两步解决conda安装pytorch时下载速度慢or超时的问题_anaconda下载pytorcg太慢-CSDN博客

虽然博主的操作系统是Ubuntu,但是中间操作应该也可以在Windows系统上生效。 

YOLOv8安装

YOLOv8官方文档:快速入门 -Ultralytics YOLOv8 文档

 YOLOv8项目地址:【wpsshop博客】

推荐阅读
相关标签