- 导出conda安装使用命令conda list -e > requirements.txt
- 导入使用conda install --yes --file requirements.txt
- 导出pip安装使用命令pip freeze > requirements.txt
- 导入使用pip install -r requirements.txt
- pytorch 1.7.1(CUDA 10.2) 使用:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch
- tensotboardX 2.4.1 方便记录日志情况, 使用:
pip install tensorboardX
- torchsummary 1.5.1 方便测试模型大小情况, 使用:
pip install torchsummary
- pynvml 11.4.1 处理显卡相关参数的信息, 使用:
pip install pynvml
- tqdm 4.62.3 使用进度条来计算时间, 使用:
pip install tqdm
- 依赖platform, argparse等基础库
其余参考
requirements.txt
以及
profile&store 下 pip-requirements.txt && conda-requirements.txt
GNU Lesser General Public License v2.1