做AI开发时,经常需要用到github上的各种代码,但是这些代码可能需要不同版本的AI开发框架和不同版本的CUDA、cuDNN才能跑通,于是我们的机器上可能要安装多个版本的CUDA和cuDNN,那么如何查看当前机器上已安装的CUDA和cuDNN呢?
1. 查看CUDA版本
cuda默认安装在/usr/local目录,可以使用ls -l /usr/local | grep cuda查看该目录下有哪些cuda版本
假设有如下输出:
lrwxrwxrwx 1 root root 8 Apr 26 cuda -> cuda-9.0
drwxr-xr-x 11 root root 4096 Apr 26 cuda-10.0
drwxr-xr-x 11 root root 4096 Apr 26 cuda-8.0
drwxr-xr-x 11 root root 4096 Apr 26 cuda-9.0
则表示当前机器上安装了8.0、9.0、10.0三个cuda版本,/usr/local/cuda是一个软链接,链接到了/usr/local/cuda-9.0目录,表示当前使用的是cuda-9.0版本。
如果要查看详细的cuda版本号,可以用如下两种方法:
1)nvcc --version,如果提示找不到该命令,则执行/usr/local/cuda/bin/nvcc --version,如果报找不到该路径或文件,则表示nvcc没有安装,可以sudo apt install nvidia-cuda-toolkit安装
2)cat /usr/local/cuda/version.txt
如果当前机器上安装了多个c