cuda版本查看_CUDA版本检测

本文介绍了CUDA版本的检测方法,包括通过版本文件、nvcc命令、代码方式以及检查驱动和环境变量设置。同时,针对nvcc路径问题进行了详细解释,提供了解决方案。
摘要由CSDN通过智能技术生成

当CUDA9安装完成后,需要进行检测:

1) cat /usr/local/cuda/version.txt #这种方法不太准,得到9.0.176

2) nvcc --version #这种方法准。如果nvcc没有安装,则sudo apt install nvidia-cuda-toolkit。检查出来是7.5.17

3)通过代码来查:

cudaDriverGetVersion(&driver_version); #获取cuda版本。得到9.0

cudaRuntimeGetVersion(&runtime_version); #获取cuda运行时版本。得到7.5

这种方法,例子是https://blog.csdn.net/langb2014/article/details/51348501

还有一个中文的例子https://www.2cto.com/kf/201708/666635.html

其中,CMakeLists.txt的写法,例子是https://blog.csdn.net/jz4626828/article/details/79878945

4)检测显卡型号及NVIDIA驱动版本

查看GPU型号 lspci | grep -i nvidia 得到M4000

查看NVIDIA驱动版本 sudo dpkg --list | grep nvidia-* 得到384.130

根据不同版本cuda对应的NVIDIA驱动版本(不同版本cuda对应的NVIDIA驱动版本 - LearnFromNow - 博客园),这个驱动应该对应cuda9.0

5)进一步查是否环境变量的问题,参见 Ubuntu16.04下安装多版本cuda和cudnn - tunhuzhuang1836的博客 - CSDN博客

在~/.bashrc中查看有无类似的语句,如有改成下面的样子,没有则增加:

export PATH="$PATH:/usr/local/cuda/bin"

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"

export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda/lib64"

然后继续nvcc --version还是得到7.5。

6)检查nvcc

which nvcc,发现是/usr/bin/nvcc,而实际上nvcc是应该在/usr/local/cuda/bin/nvcc。

所以,sudo gedit /usr/bin/nvcc,把里面的内容"exec /usr/lib/nvidia-cuda-toolkit/bin/nvcc" 改成"exec /usr/local/cuda/bin/nvcc"

然后继续nvcc --version这次得到9.0

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值