Win10配置YOLO目标检测Pytorch环境

系统环境：Win10+显卡3090

安装cuda和cudnn：

Windows10系统pytorch、cuda11.0、cuDNN安装

注意事项：

1.查看显卡版本后下载对应版本的cuda和cudnn，链接如下：

cuda下载：https://developer.nvidia.com/cuda-toolkit-archive

cudnn下载：https://developer.nvidia.com/rdp/cudnn-archive

2.安装好cuda后，安装cudnn即将下载的压缩包里的lib,include,bin文件添加到cuda对应路径下。

注意，一台电脑可以安装多个版本的cuda,版本之间的切换只需要将系统环境变量对应版本的路径前置即可。同时，执行Pytorch官方的conda安装命令，也可以在anaconda创建的虚拟环境中搭建pytorch所需对应版本的cuda和cudnn。另外，可以使用conda命令在虚拟环境中单独安装所需cuda和cudnn：例如conda install cudatoolkit=11.2 cudnn=8.1

安装Anaconda:

win10中安装Anaconda

安装Pytorch：

Pytorch官网：https://pytorch.org/

创建虚拟环境：conda create -n env_name python=3.9

注意事项：除非自己有编译需求，否则无需先安装cuda和cudnn，根据官网，只需一条命令即可。因此可以任意选择需要的Pytorch版本，不一定要安装显卡驱动对应的cuda版本，因为Pytorch官网conda命令安装pytorch时会在虚拟环境中重新配置对应的cuda和cudnn。

conda添加清华、阿里源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/main

conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/r

conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/msys2

conda config --set show_channel_urls yes

验证pytorch、cuda、cudnn：

python
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)
print(torch.cuda.is_available())

torch.version.cuda

torch.backends.cudnn.is_available()

torch.backends.cudnn.version()

安装Opencv:

conda install -c conda-forge opencv

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-contrib-python
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python

pip添加清华源：

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

安装pycocotools:

pip install cython

（Linux）pip install pycocotools或者（Win）pip install pycocotools-win

Windows和Linux端简单安装pycocotools方法