ubuntu查看cuda、cudnn版本

原创已于 2022-05-15 21:35:30 修改 · 2.2k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #深度学习 #linux

于 2022-05-15 20:14:01 首次发布

tensorflow 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了如何查看和切换CUDA及CUDNN版本。通过`cat /usr/local/cuda/version.txt`可以查看CUDA版本，而`cat /usr/local/cuda/include/cudnn.h|grep CUDNN_MAJOR -A2`则用于查看CUDNN版本。若需切换CUDA版本，可删除原有软链接并创建新的指向所需版本的链接。同样，切换CUDNN版本需要解压新版本文件，并将头文件和库文件替换到CUDA相应目录下，最后更新权限。这些步骤对于深度学习和GPU加速计算的环境配置至关重要。

查看cuda 版本

cat /usr/local/cuda/version.txt

查看cudnn 版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

cuda版本切换

rm -rf /usr/local/cuda #删除之前的软连接
sudo ln -s /usr/local/cuda-10.0/ /usr/local/cuda/
nvcc --version #查看当前nvcc版本

cudnn版本切换

sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*