一键配置Ollama环境变量 程序
时间: 2025-07-08 18:47:57 浏览: 3
### 实现一键配置Ollama环境变量
为了实现一键配置Ollama环境变量的功能,可以编写一个Python脚本来自动完成这一过程。该脚本会根据操作系统的不同采取不同的方法来设置环境变量。
对于Linux/Unix系统:
```python
import os
import subprocess
def set_env_variable_linux(path):
"""为 Linux 设置 OLLAMA_MODELS_PATH 环境变量"""
service_file_path = "/etc/systemd/system/ollama.service"
with open(service_file_path, 'a') as file:
file.write(f'Environment=OLLAMA_MODELS_PATH={path}\n')
subprocess.run(['systemctl', 'daemon-reload'])
subprocess.run(['systemctl', 'restart', 'ollama'])
if __name__ == "__main__":
model_path = input("请输入要设置的模型路径:")
set_env_variable_linux(model_path)
```
对于Windows系统,则可以通过修改注册表的方式快速设定环境变量:
```python
import winreg as reg
from tkinter import Tk
from tkinter.filedialog import askdirectory
def add_to_registry(key_name,value):
try:
key = reg.OpenKey(reg.HKEY_LOCAL_MACHINE,r'SYSTEM\CurrentControlSet\
\Control\Session Manager\Environment',0,reg.KEY_ALL_ACCESS)
reg.SetValueEx(key,key_name,0,reg.REG_EXPAND_SZ,value)
reg.CloseKey(key)
print('成功添加到系统环境变量.')
except Exception as e:
print(e)
if __name__ == '__main__':
root = Tk()
root.withdraw() # 隐藏tkinter的小窗口
path = askdirectory(title='选择文件夹') # 弹出对话框让用户选择文件夹
if not path.strip():
exit()
add_to_registry("OLLAMA_MODELS", path)
print("请重启计算机以应用更改.")
```
上述代码片段展示了针对两种操作系统的一键配置方案[^1]。需要注意的是,在实际部署前应当测试这些脚本并确保其安全性以及兼容性。
阅读全文
相关推荐

















