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

Question

我想使用 __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() {}

我不知道要传旗

Answer 1

我对 relevant code 的阅读表明你不能。似乎不支持通过该插件向 nvcc 传递额外的参数。

如果您不明白如何操作，您将需要分叉和修改插件来尝试或请求插件的作者做同样的事情。

或者，使用 ，强制插件写入文件，然后由 运行 nvcc 自己手动编译该文件。