Mr_老冷 发表于 2024-8-29 18:22:19

【YOLO】YOLOv8 GPU版显卡CUDA环境安装与配置

前置条件:
Nvidia英伟达显卡
推荐4G显存以上,如1050ti
最低要求750ti
升级显卡驱动到最新
//player.bilibili.com/player.html?isOutside=true&aid=113049142822768&bvid=BV1b3HLeTEAB&cid=25651578697&p=1
图文参考:https://blog.csdn.net/shangyanaf/article/details/139029717

yolo环境包
百度网盘
老冷网盘

环境变量配置


C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\lib\x64



torch包安装代码
先把torch包放到指定目录,然后在Anaconda的cmd控制台中切换到该目录
pip install torch-1.13.1+cu116-cp38-cp38-win_amd64.whl
pip install torchaudio-0.13.1+cu116-cp38-cp38-win_amd64.whl
pip install torchvision-0.14.1+cu116-cp38-cp38-win_amd64.whl
安装后5个包即可全部删除

页: [1]
查看完整版本: 【YOLO】YOLOv8 GPU版显卡CUDA环境安装与配置