如何在google colab中将标志传递给nvcc？答案

【问题标题】：How to pass flags to nvcc in google colab?如何在google colab中将标志传递给nvcc？
【发布时间】：2021-04-21 17:12:59
【问题描述】：

我想使用__device__ lambda，但它说我需要传递标志'--expt-extended-lambda'。

我在 colab 中使用这个环境。

!wget https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda-repo-ubuntu1604-9-2- 
local_9.2.88-1_amd64 -O cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb
!dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb
!apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub
!apt-get update
!apt-get install cuda-9.2
!nvcc --version

!pip install git+git://github.com/andreinechaev/nvcc4jupyter.git
%load_ext nvcc_plugin

我写这样的代码

%%cu
int main() {}

我不知道要传旗。

【问题讨论】：

标签： jupyter-notebook cuda google-colaboratory nvcc compiler-flags

【解决方案1】：

我对@987654321@ 的阅读表明您不能。似乎不支持通过该插件向 nvcc 传递其他参数。

如果您不了解如何执行此操作，则需要分叉和修改插件以品尝或请求插件作者执行相同操作。

或者，使用this hack，它会强制插件写入文件，然后通过自己运行 nvcc 手动编译该文件。

【讨论】：