几个环境下的cuda版本

最近在搭建mmdetection3d环境的时候，报了一个cuda相关的错误"CUDA Toolkit 10.2 is unsupported by Numba - 11.2 is the minimum required version"，但是我本地安装的cuda版本是11.3，按道理说不会出现检测到10.2的版本。后来弄懂了，原因是讲cuda版本是要分几个环境的。这个相似的问题包括torch.version.cuda版本与本地安装的cuda版本不一致。

1、驱动环境

如果有安装好NVIDIA驱动，ubuntu上可以执行命令：

nvidia-smi

显示的cuda版本是当前显卡驱动下支持的最高cuda版本，也就是说在nvidia官网下载安装的cuda版本以及用conda安装的cudatoolkit版本不能超过该版本。

2、本地cuda安装的环境

这个环境下的cuda是通过在NVIDIA官网下载安装的cuda，安装好之后输入如下命令就可以看到cuda版本：

nvcc -V

安装好之后有cuda相关的头文件、lib、dll文件，据我所知，一般在两种场景下才用得到：

1）在用c++写cuda程序，以及跑cuda程序的时候用到；

2）在深度学习模型部署端会用到，比如cudnn/onnxruntime/mnn等会依赖cuda包

3、conda环境

这个环境下一般是在用conda安装深度学习训练库的时候用得到（比如pytorch/tensorflow），conda安装命令如下:

conda install cudatoolkit

安装好后可以通过如下命令查看cuda版本：

conda list | grep cuda

如果有安装pytorch，则可以通过如下命令查看cuda版本：

python -c 'import torch;print(torch.version.cuda)'

如果有安装numba，则可以通过如下命令查看cuda版本:

python -c 'from numba.cuda.cudadrv.runtime import runtime;print(runtime.get_version())'