python下安装cuda相关包时报错CUDA_TOOLKIT_ROOT_DIR must be defined

最近在用pycuda编程时想要用一个支持python的kmeans并行化包，github链接在这里src-d/kmcuda, 里边有详细的安装教程。但我按照上边提供的安装方法操作之后遇到了错误：

CUDA_TOOLKIT_ROOT_DIR must be defined

此时需进行的操作是：
打开.bashrc文件，位置在home下，处于隐藏状态，可以通过指令：

gedit ~/.bashrc  直接打开

也可以在home下按Ctrl+H把该文件显示出来，选择在终端下打开。
打开之后在文件中添加CUDA_TOOLKIT的路径，比如我的是：

export CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda10.0

保存之后即可解决这个错误。