使用optirun在虚拟环境中运行python脚本

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (1)
  • 关注 (0)
  • 查看 (69)

如果工具需要专用的GPU(只能通过大黄蜂或者可用),那么如何运行使用虚拟环境的python工具和像click这样的库?optirun

例:

#normally (with nvidia driver being the default)
agent --run-with-tf-gpu train

agent脚本触发基于单击的应用程序,然后使用该venv/bin/python二进制文件。

我有一个解决方案(见下面的答案),但我很高兴用“正确答案”奖励更好的解决方案

提问于
用户回答回答于

选项1:用包装器替换二进制文件

p=`pwd`
cd venv/bin
rm python #soft link to python3.6
echo 'optirun $p/venv/bin/python3.6 "$*"' > python
chmod +x python

选项2:创建别名

只有在没有使用绝对路径调用python时才有效。

venv/bin/activate文件末尾插入这三行:

#setting optirun to be used when running python in venv
alias python="optirun python"
alias python3="optirun python3"

扫码关注云+社区

领取腾讯云代金券