python - 如何在 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 检测到我的用户环境变量?是否有我需要修改的特定路径?
谢谢。
解决方案
像这样
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 文件