使用Pycharm本地模型进行迁移的时候,Pycharm的虚拟环境依赖包需导出成一个文件requirement.txt,方便给他人或者模型部署时模型环境搭建。
方法一:终端命令行方式
Linux或cmd中的一般命令:
pip freeze > requirements.txt
方法二:python脚本方式生成
但以上命令直接在Windows终端是无法使用:所以你需要使用以下代码,进行导出,将如下代码新建一个Python文件,放在项目根路径下,然后运行即可生成我们所需的requirments.txt 文件。
Windows和Linux下均可正常生成。
import os
import platform
import sys
# find pwd
project_root = os.path.dirname(os.path.realpath(__file__))
print('current path' + project_root)
# code for Linux or Windows system
if platform.system() == 'Linux':
command = sys.executable + ' -m pip freeze > ' + project_root + '/requirements.txt'
if platform.system() == 'Windows':
command = '"' + sys.executable + '"' + ' -m pip freeze > "' + project_root + '\\requirements.txt"'
print(command)
# execute
os.popen(command)