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

问题描述

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

我不知道要通过标志。

解决方法

我对 relevant code 的阅读表明你不能。似乎不支持通过该插件将附加参数传递给 nvcc。

如果您不了解如何执行此操作，则需要对插件进行分叉和修改以品尝或请求插件的作者也这样做。

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