Question

我想在带有NVIDIA Optimus（bumblebee）的Linux笔记本电脑上运行来自IPython笔记本的pycuda。通常，我可以通过键入optirun python my_pycuda_script.py

来运行python脚本

但是，如果我启动optirun ipython notebook然后打开一个笔记本，一个新的内核启动，我不能再运行pycuda了。我发现，如果我用调用optirun new_location_of_python的shell脚本替换我的python可执行文件它正在运行 - 但这是一个非常难看的黑客。有一个更好的方法吗？也许有一个神奇的功能，所以只有相关的笔记本电脑是用optirun启动的？

感谢您的帮助！

Answer 1

我刚从github:data_science_workspace找到了解决方案。

对Jupyter的GPU支持：

对于使用optimus的linux上的计算机，你必须创建一个将使用＆＃34; optirun＆＃34;能够使用GPU加速。为此，请转到以下文件夹：

cd ~/.local/share/jupyter/kernels/

然后编辑文件python3/kernel.json，以便将"optirun"添加为argv数组中的第一个条目：

{
"language": "python",
"display_name": "Python 3",
"argv": [
    "optirun",
    "/home/fabien/.conda/envs/data_science/bin/python",
    "-m",
    "ipykernel",
    "-f",
    "{connection_file}"
    ]
}

但在我的计算机中，kernel.json位于：~/miniconda3/envs/nn/share/jupyter/kernels/python3。

我的康茄达信息：

$ conda info
user-agent : conda/4.3.30 requests/2.14.2 CPython/3.6.1 Linux/4.9.79-1-MANJARO arch/Manjaro glibc/2.26

希望这就是你所需要的： - ）。

带有optirun的IPython笔记本

1 个答案: