我最近发现了一个用于此目的的脚本,这里是代码的链接。
conda_env_export.py
由于代码很短,我可以粘贴下面的代码。请注意,我修改了一些地方供我个人使用。
import re
import subprocess
import sys
import yaml
import fire
CONDA_PATH = '/home/ubuntu/miniconda3/condabin/conda'
def export_env(history_only=False):
cmd = [CONDA_PATH, 'env', 'export']
if history_only:
cmd.append('--from-history')
cp = subprocess.run(cmd, stdout=subprocess.PIPE)
try:
cp.check_returncode()
except:
raise
else:
return yaml.safe_load(cp.stdout)
def _is_history_dep(d, history_deps):
if not isinstance(d, str):
return False
d_prefix = re.sub(r'=.*', '', d)
return d_prefix in history_deps
def _get_pip_deps(full_deps):
for dep in full_deps:
if isinstance(dep, dict) and 'pip' in dep:
return dep
def _combine_env_data(env_data_full, env_data_hist):
deps_full = env_data_full['dependencies']
deps_hist = env_data_hist['dependencies']
deps = [dep for dep in deps_full if _is_history_dep(dep, deps_hist)]
pip_deps = _get_pip_deps(deps_full)
env_data = {}
env_data['channels'] = env_data_full['channels']
env_data['dependencies'] = deps
env_data['dependencies'].append(pip_deps)
return env_data
def main(s_save=None):
env_data_full = export_env()
env_data_hist = export_env(history_only=True)
env_data = _combine_env_data(env_data_full, env_data_hist)
if s_save:
with open(s_save, 'w') as y:
yaml.dump(env_data, y)
else:
yaml.dump(env_data, sys.stdout)
if __name__ == '__main__':
fire.Fire(main)
教程
将以上代码另存为conda_export_minimal.py。
用你的 conda 路径指定CONDA_PATH(可以通过运行命令whereis conda 找到)。
然后运行如下:
python conda_export_minimal.py --s_save=env.yml
一个新文件env.yml 将出现在您的当前目录中,其中包含关于 conda 和 pip 的最小依赖项。