有没有快速的命令或脚本来检查安装的CUDA版本?
我在安装目录下找到了4.0的手册,但我不确定它是否是实际安装版本。
如果nvcc和nvidia-smi之间存在版本不匹配,则会使用不同版本的cuda作为驱动程序和运行时环境。
为确保使用相同版本的CUDA驱动程序,您需要将CUDA添加到系统路径中。
首先运行whereis cuda并找到cuda驱动程序的位置。
然后进入.bashrc文件并修改路径变量,使用变量“LD_LIBRARY_PATH”设置搜索目录的优先顺序。
例如
$ whereis cuda
cuda: /usr/lib/cuda /usr/include/cuda.h /usr/local/cuda
vim ~/.bashrc
export PATH="/usr/local/cuda/bin:${PATH}"
在此行后,将目录搜索路径设置为:
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"
然后保存 .bashrc 文件。并刷新它:
$ source ~/.bashrc
这将确保您拥有nvcc -V和nvidia-smi以使用相同版本的驱动程序。
nvcc --version
命令会提示The program 'nvcc' is currently not installed. You can install it by typing: sudo apt install nvidia-cuda-toolkit
,但是nvidia-smi
命令显示 CUDA 版本为 10.1。 - mrgloomcat /usr/local/cuda/version.txt
命令可以得到更精确的版本信息:CUDA 版本 10.1.168
。 - mrgloomnvidia-smi
混合获取版本,使用grep
识别行,并使用sed
删除不必要的字符:nvidia-smi | grep -o 'CUDA Version: [0-9].\.[0-9]' | sed 's/.*: //'
。 - Eduardo Pignatelli