Python 在google colab上安装pycuda

我想在google colab上安装pycuda。 我尝试了

pip安装pycuda

，但它给出了

警告：nvcc不在路径中。可能需要设置CUDA_INC_DIR才能成功安装

在读了一些博客之后，我也做了

！导出路径=/usr/local/cuda/bin:$PATH

和

！导出CUDA_ROOT=/usr/local/CUDA

，但仍然会给出相同的错误

此外，在终止安装时，还显示

In file included from src/cpp/cuda.cpp:1:0:
src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory
#include <cuda.h>
          ^~~~~~~~
compilation terminated.
error: command 'x86_64-linux-gnu-gcc' failed with exit status 1

包含在src/cpp/cuda.cpp:1:0中的文件中：
src/cpp/cuda.hpp:14:10:致命错误：cuda.h:没有这样的文件或目录
#包括
^~~~~~~~
编译终止。
错误：命令“x86_64-linux-gnu-gcc”失败，退出状态为1

此脚本还安装了

pycuda

所需的软件包：

!curl https://colab.chainer.org/install | sh -

---

你能加上所有的错误吗？