yolov5环境搭建与pytorch中torch、torchvision、torchaudio安装

瓶困小张

已于 2023-07-03 22:06:45 修改

阅读量4.1k

点赞数

分类专栏：安装包文章标签：深度学习 python pytorch

于 2022-11-07 17:16:33 首次发布

原文链接：https://blog.csdn.net/shiwanghualuo/article/details/122860521

版权

安装包专栏收录该内容

1 篇文章 0 订阅

订阅专栏

python软件安装（可以不单独安装）

(2条消息) Python安装教程（2022最新）_学Python的阿杜的博客-CSDN博客

安装anaconda3:

2020.2对应的python版本为3.7(不推荐)

2020.11对应的python版本为3.8(推荐python3.8及以上,但是尽量不要用最新的,存在兼容问题)

注:添加环境变量时，没有user这一文件夹不影响

~~~~~~~~~

~~~~~~~~~

对于新电脑，可以直接安装最新的anaconda版本，python为3.10，cuda安装（pytorch适用于的最高的）11.8版本

Anaconda超详细安装教程（Windows环境下）_conda安装_菜鸟1号！！的博客-CSDN博客

卸载原有的anaconda并安装

(2条消息) Anaconda||（踩坑无数，含泪总结！！！）Anaconda的卸载与安装（tensorflow+Keras+spyder+添加镜像源）_anaconda装到一半不动了_Inochigohan的博客-CSDN博客

里面还有tensorflow环境的安装

anaconda安装的环境默认c盘的解决方法：

Anaconda 安装及修改环境默认位置_anaconda默认安装路径_学习纪实的博客-CSDN博客

无法枚举，拒绝访问时。

修改高级选项，修改权限everyone时记得勾选下面的框

(4条消息) 修改添加用户权限，提示“无法枚举容器中的对象访问被拒绝” 解决方法_无法枚举容器中的对象,访问被拒绝_赵氏天儿的博客-CSDN博客

安装pycharm

(36条消息) pycharm安装教程，超详细_皮小孩ls的博客-CSDN博客

加速器配置:

(4条消息) win10深度学习环境搭建_江南小赣的博客-CSDN博客

cuda加速器版本选择:Release Notes :: CUDA Toolkit Documentation (nvidia.com)

cudun下载:cuDNN Archive | NVIDIA Developer

~~~~~~~~~

GPU版本:注意,要完整添加好环境变量(4条消息) GPU版本的PyTorch安装与环境配置_「已注销」的博客-CSDN博客_安装gpu版本的pytorch

~~~~~~~~~

安装cuda需要添加的环境变量：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\CUPTI\lib64

pytorch环境搭建及使用

打开anaconda安装环境界面

查看有哪些环境：conda env list

创建一个新环境：conda create -n pytorch python=3.8 （注：pytorch为环境的名字，python版本为3.8）

询问是否安装，输入y

激活/进入环境：conda activate pytorch

“在Anaconda prompt中操作：增加配置包的速度
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
查看是否修改好通道：

conda config --show channels”

注：如果要删除自定义源更换回conda的默认源，直接删除channels即可，命令如下：
conda config --remove-key channels

pytorch中torch、torchvision、torchaudio安装

1、gpu- torch

（推荐）安装方法①直接去pytorh官网（https://pytorch.org/）

选择型号

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

（118代表cuda为11.8，安装时间会较久）

安装时出现问题：ERROR: Could not install packages due to an OSError: [WinError 5] 拒绝访问。: 'D:\\Anaconda3\\envs\\Y5\\Lib\\site-packages\\~orch\\lib\\torch_cuda_cu.dll' Consider using the `--user` option or check the permissions.

解决：找到“D:\\Anaconda3\\envs\\Y5”文件夹，在此文件夹上点鼠标右键-属性-安全

(1条消息) 关于PermissionError: [WinError 5] 拒绝访问的解决方案_winerror5 拒绝访问_@Liu_GuoXing的博客-CSDN博客

接下来输入python，进入python编辑器，

输入import torch测试pytorch是否已经安装成功，

再输入torch.cuda.is_available()查看cuda能否调用GPU”

输入exit()退出

下载文件式安装pytorch（不推荐）

安装方法②下载whl文件进行配置

对应关系:

(4条消息) PyTorch中torch、torchvision、torchaudio版本对应关系_Liekkas Kono的博客-CSDN博客_torchaudio版本

python3.8使用的是以下版本（版本一定要对应）

如何安装torch、Pytorch、torchversion_安装torchversion_-Han-的博客-CSDN博客

torch系列网站下载文件放置在桌面上，链接:

https://download.pytorch.org/whl/torch_stable.html

GPU版本为cu*torch，下载win，cp后面跟着是python版本

安装教程 :

(4条消息) 如何安装torch、Pytorch、torchversion_-Han-的博客-CSDN博客_torch.version

输入cd Desktop

输入pip install +文件名.whl

（如pip install torchvision-0.13.1+cu113-cp38-cp38-win_amd64.whl

代表torchvision-0.13.1，我的cuda为11.3，python为3.8，我的系统为Windows64位）

出现的问题：使用 conda list查看此环境，里面出现很多orge

解决：使用conda卸载Pytorch（但是环境也乱了）

输入conda uninstall pytorch
conda uninstall libtorch

2、cpu- torch（运行太慢了，不推荐）

直接去pytorh官网

GPU版本的PyTorch安装与环境配置_gpu pytorch_「已注销」的博客-CSDN博客

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

conda install pytorch torchvision cudatoolkit=11.8

yolov5中其他一些包的安装

matplotlib安装：pip install matplotlib

tqdm安装：pip install tqdm

yaml安装：pip install pyyaml

scipy安装：pip install scipy

tensorboard安装：pip install tensorboard

seaborn安装：pip install seaborn

thop安装：pip install thop

opencv安装：pip install opencv-python

最麻烦的是pycocotools：添加个coco数据集就行（不行还有下列两种解决方法）

~~~~~~~~

其他代码可能需要安装的包：

scikit-learn安装：pip install scikit-learn

opencv安装：pip install opencv-python

问题1：关于pycocotools安装（添加数据集后也没用）：subprocess.CalledProcessError: Command 'pip install 'pycocotools>=2.0'' returned non-zero exit status 1.

解决方法1：conda install -c esri pycocotools （只适用于python<=3.8）

(1条消息) 2020年最新实测pip安装pycocotools失败的解决办法_no matching distribution found for pycocotools-win_wtandyn的博客-CSDN博客

(1条消息) Win10复现yolov5时有关pycocotools＞=2.0错误的解决办法_天蝎美男oldwang的博客-CSDN博客

解决方法2：安装visual studio（适用范围广）

(18条消息) pycocotools安装_soga_Moji的博客-CSDN博客

问题2：额外安装一些环境包时，出现：OSError: [WinError 127] 找不到指定的程序。 Error loading "D:\Anaconda3\envs\Y5\lib\site-packages\torch\lib\c10_cuda.dll" or one of its dependencies.

待解决

pycharm中设置环境

注：选择Make available to all projects则每个打开的文件夹中都可以直接在右下角选择此环境，不需再重复进行此操作，并且环境编号不会改变（例如出现‘（1）’）。

新版pycharm会出现一个问题：pycharm 添加Anaconda中的虚拟环境时，python解释器出现Conda executable is not found

(3条消息) PyCharm添加Anaconda中的虚拟环境，Python解释器出现Conda executable is not found_努力向前的JF（s1hjf）的博客-CSDN博客

解决方法1：1、选择Library\bin下的conda.bat——点击右侧 load environment出现以下界面

2、选择创造的环境（但是环境里面的安装包还是显示不全）

错误的解决：

错误1：AttributeError: module 'distutils' has no attribute 'version'

AttributeError: module ‘distutils‘ has no attribute ‘version‘ 解决方案 - 知乎 (zhihu.com)

原因分析：setuptools版本问题”，版本过高导致的问题。

解决方法：pip uninstall setuptools

conda install setuptools==58.0.4

a poor girl暴雨哭泣T^T，羞涩的口袋告诉我，我得自己装