linux如何检查cudnn安装成功
时间: 2023-08-09 19:12:39 浏览: 347
要检查CUDNN是否成功安装在Linux上,可以执行以下步骤:
1. 打开终端窗口,输入以下命令来检查CUDNN库文件是否存在:
```
ls /usr/local/cuda/include/cudnn.h
```
如果文件存在,则表示CUDNN已经成功安装。
2. 检查CUDA和CUDNN的版本是否匹配。执行以下命令来查看CUDA版本:
```
nvcc --version
```
执行以下命令来查看CUDNN版本:
```
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
```
确保CUDA和CUDNN的版本兼容。
3. 编译一个简单的CUDNN示例程序来验证安装。创建一个名为`cudnn_test.cu`的文件,并将以下代码复制到文件中:
```c
#include <stdio.h>
#include <cudnn.h>
int main()
{
cudnnHandle_t cudnn;
cudnnCreate(&cudnn);
printf("CUDNN installed successfully.\n");
cudnnDestroy(cudnn);
return 0;
}
```
4. 使用以下命令编译并运行示例程序:
```
nvcc -o cudnn_test cudnn_test.cu -lcudnn
./cudnn_test
```
如果输出结果为"CUDNN installed successfully.",则表示CUDNN已经成功安装并可以正常使用。
请注意,以上步骤假设您已经正确安装了CUDA和CUDNN,并且在默认安装路径下。如果您的安装路径不同,请相应地修改命令中的路径。
阅读全文