🚀【安装CUDNN及查看版本】🚀
在进行深度学习模型训练时，选择合适的CUDA和cuDNN版本非常重要。特别是在ARM64架构下，确保正确的cuDNN版本安装对于提升性能至关重要。下面将介绍如何在ARM64架构上安装cuDNN并检查其版本。
🎯【安装步骤】🎯
1. 访问NVIDIA官网下载适用于ARM64架构的cuDNN库文件。
2. 解压下载的文件，并将包含头文件和库文件的目录复制到CUDA的安装路径中。
3. 配置环境变量，使系统能够找到新的cuDNN库文件。
🔍【查看版本】🔍
安装完成后，可以通过运行以下命令来检查cuDNN的版本：
```bash
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
```
或者使用Python代码检查：
```python
import tensorflow as tf
print(tf.__version__)
```
记得根据实际情况调整路径和命令。希望这些信息能帮助你顺利完成cuDNN的安装与版本确认！💪