【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