1. 程式人生 > >【tensorflow】檢視cuda,cudnn版本命令

【tensorflow】檢視cuda,cudnn版本命令

這兩天調系統配置,有點上火。
記錄下這兩個系統命令
cuda 版本

cat /usr/local/cuda/version.txt

cudnn 版本

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