在Ubuntu系统中xwenw.com/tag/%e6%9f%a5%e7%9c%8b" target="_blank">查看cuDNN版本,可以通过以下步骤进行:
1. 打开终端。
2. 输入以下命令,创建一个名为`cudnn_version.sh`的脚本文件:
touch cudnn_version.sh
3. 使用文本编辑器打开`cudnn_version.sh`文件,例如使用`nano`编辑器:
nano cudnn_version.sh
4. 在文件中输入以下内容:
#!/bin/bash export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64 python3 -c "import tensorflow as tf; print(tf.sysconfig.get_build_info()['cuda_compiler'])"
这里假设你的CUDA版本是9.0,如果你的CUDA版本不同,请相应地修改`/usr/local/cuda-9.0/lib64`为你的CUDA库路径。
5. 保存并关闭文件,在`nano`编辑器中,按`Ctrl + X`,然后按`Y`,最后按`Enter`。
6. 为脚本文件添加可执行权限:
chmod +x cudnn_version.sh
7. 运行脚本文件,查看cuDNN版本:
./cudnn_version.sh
输出结果类似于:
Cuda compilation tools, release 9.0, V8.0.44 for NVIDIA GeForce GTX 1080 (Build 70fdcf6)
这表明你正在使用的cuDNN版本是V8.0.44。
相关问题与解答:
1. 如何安装cuDNN?请参考NVIDIA官方文档:-started-installing-the-cudnn-archive-on-linux,按照说明进行安装即可。
2. cuDNN支持哪些CUDA版本?请参考NVIDIA官方文档:-platforms,根据你的CUDA版本选择合适的cuDNN版本进行安装。
3. 如何从源代码编译cuDNN?请参考NVIDIA官方文档:-from-source,按照说明进行编译即可。
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。
评论(0)