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方法

你可能感兴趣的:(pytorch,目标检测,深度学习)