【YOLO】YOLOv8训练环境配置 python 3.8.18 + cuda 11.3.1 + cudnn 8.2.1 + pytorch 1.12.1-gpu版 - 知乎 (zhihu.com)
一、Anaconda
默认装好了可用的Anaconda,安装教程见Win10系统+anaconda安装 - 知乎 (zhihu.com)
二、在虚拟环境下用conda安装
1.创建虚拟环境
(1)创建新的虚拟环境,使用下述语句:conda create -n yolov8 python=3.8
“yolov8”是我给虚拟环境起的名字
(2)激活虚拟环境,输入activate yolov8
2.安装cuda+cudnn
(1)先配置镜像网址(参照https://mirror.tuna.tsinghua.edu.cn/help/anaconda/):
进入anaconda prompt->依次运行以下语句:
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
conda config --set show_channel_urls yes
2)查看conda可供安装版本
运行conda search cudatoolkit
同样运行conda search cudnn
看到用conda install可以安装的各种cudatoolkit和cudnn版本以及匹配关系,确认好自己需要的版本。我用的1080Ti显卡,选择了cudatoolkit=11.3.1和cudnn=8.2.1
(3)装cudatoolkit
输入conda install cudatoolkit=11.3.1,敲y
(4)装cudnn
输入conda install cudnn=8.2.1,敲y