【tensorflow】查看cuda,cudnn版本命令

来源:互联网 发布:带约束条件的遗传算法 编辑:程序博客网 时间:2024/06/08 13:53

这两天调系统配置,有点上火。
记录下这两个系统命令
cuda 版本

cat /usr/local/cuda/version.txt

cudnn 版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
原创粉丝点击