首页 > 解决方案 > 如何在 Google Colab 中设置环境变量?

问题描述

在我的本地计算机上,我可以简单地进入“系统属性 -> 环境变量”并在用户变量中添加一个新变量及其值。

然后,我可以通过在 Python 中使用它来检索这个值:

import os
os.environ["VAR_NAME"]

但是,我最近才开始使用 Google Colab,它似乎无法检测到环境变量,因为它给了我这个错误:

---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-36-28128554cf91> in <module>()
      1 import os
----> 2 os.environ["REDDIT_NAME"]

/usr/lib/python3.7/os.py in __getitem__(self, key)
    679         except KeyError:
    680             # raise KeyError with the original key value
--> 681             raise KeyError(key) from None
    682         return self.decodevalue(value)
    683 

KeyError: 'REDDIT_NAME'

我应该怎么做才能让 Google Colab 检测到我的用户环境变量?是否有我需要修改的特定路径?

谢谢。

标签: pythonenvironment-variablesgoogle-colaboratory

解决方案


像这样

import os
os.environ['REDDIT_NAME'] = 'something'
print(os.getenv('REDDIT_NAME'))

或者使用dotenv库。将环境保存在文件中:

  import dotenv
  dotenv.load_dotenv(os.path.join(os.path.dirname(__file__), './.env'))

在 .env 文件中: REDDIT_NAME = something

推送到 git 时忽略 .env 文件


推荐阅读