赞
踩
硬件环境:实验室集群-显卡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官方文档:快速入门 -Ultralytics YOLOv8 文档
YOLOv8项目地址:【wpsshop博客】
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。