yolov8训练的一些工具类代码以及环境包

本文介绍了如何在Windows上安装GPU环境,包括检查CUDA版本、安装Python、确保pip存在并升级,以及使用PyTorch的官方和镜像站快速全局安装torch、torchvision和torchaudio的过程。

gitee导航
直接下载自己改,源码我指不定哪天就会按我自己的训练修改

1.gpu环境安装

cmd执行
nvidia-smi

在这里插入图片描述
主要看下cuda的版本的显存大小
在这里插入图片描述
上面标记的是我下载下来能跑起来的版本

python 选择 Windows installer (64-bit),下载安装,勾选添加环境
地址:https://www.python.org/downloads/release/python-3116/

cmd查看是否有pip或者pip3, 没有的话cmd执行
python -m ensurepip --upgrade

cuda安装,选个自己显卡支持的版本,全部点下一步,装C盘蟹蟹!!
https://developer.nvidia.com/cuda-toolkit-archive

torch官网安装,一般会很慢,要么翻墙要么自己先去下面的http链接下载,全局安装就行了
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

cuda网站进入卡可能要翻墙,这里推荐下我使用的还不错的,你们可以尝试

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值