【Python】使用 requirements.txt 与 pytorch 相关配置
前言
进行深度学习时，学习github上面的项目，经常需要配置不同的虚拟环境。最简便的方式是使用 requirements.txt 导入所需的项目，但是涉及到 pytorch 时，需要特别注意。如果你的 requirements.txt 文件中的 PyTorch 版本与你的环境不兼容，你可能会在安装时遇到错误。在这种情况下，你需要手动指定一个与你的环境兼容的 PyTorch 版本。
一、pip
1、导出结果含有路径
导出结果会存在路径，生成的 requirements.txt 文件在当前目录下。
pip freezen > requirements.txt
2、导出不带路径的
生成的requirements.txt文件在当前目录下。
pip list --format=freeze >requirement.txt
生成 requirements.txt，pip freeze 会将当前PC环境下所有的安装包都进行生成,再进行安装的时候会全部安装很多没有的包，此方法需要注意。
二、Conda
1、导出requirements.txt
a. 导出
conda list -e > requirements.txt
b. 导入安装
conda install --yes --file requirements.txt
注意：使用 requirements.txt 导入时，如果有 pytorch 或者torch 先删除 requirements.txt 里面的的pytorch，安装其它库起
2、导出yml 文件
a.导出
conda env export > freeze.yml
b.导入安装
conda env create -f freeze.yml
三、第三方包：pipreqs（推荐）
使用pipreqs，这个工具的好处是可以通过对项目目录的扫描，发现使用了哪些库，生成依赖清单。
step1：安装pipreqs(默认没有安装)
pip install pipreqs
step2：使用pipreqs导出
在python项目的根目录下 使用
pipreqs ./
如果报错，则采用下面的代码
pipreqs ./ --encoding=utf-8
生成的requirements.txt文件在当前目录下。
1、创建并激活conda环境
conda create -n 环境名称 python=3.10 anaconda
conda activate 环境名称
2、安装requirements文件的pip源的包
cd 到 requirements.txt 文件所在的路径下，然后使用国内镜像网站安装
pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ -r requirements.txt
四、pytorch的配置
注意 PyTorch 版本和兼容性：
requirements.txt 文件中列出的 pytorch 版本可能与你的系统环境（如操作系统、Python版本、CUDA版本等）不兼容。PyTorch 有许多不同的构建版本，对应于不同的操作系统、Python版本和CUDA版本。如果你的 requirements.txt 文件中的 PyTorch 版本与你的环境不兼容，你可能会在安装时遇到错误。在这种情况下，你需要手动指定一个与你的环境兼容的 PyTorch 版本。
由于 python 版本和pytorch和torchvision和cuda有对应的关系，需要明确项目使用上面版本进行下一步的conda虚拟环境的创建。这里使用当今最火爆的yolov5项目的requirement，作为示例，如下图所示，所要求的pytorch版本为python>=3.7.0 torch>=1.7.0 torchvision>=0.8.1
安装正确版本的pytorch+torchvision+cuda
具体的版本的下载代码可以在这里查到，这里举例使用国内镜像安装，仅在原始命令后，添加下载网址。
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/