1. 程式人生 > 實用技巧 >檢視Tensorflow、CUDA及cuDNN版本

檢視Tensorflow、CUDA及cuDNN版本

檢視Tensorflow、CUDA及cuDNN版本

1. 檢視Tensorflow版本

開啟cmd,輸入
python
import tensorflow as tf
tf.__version__
在這裡插入圖片描述

2. 檢視windows的CUDA版本

法一:開啟cmd,輸入 nvcc --version
在這裡插入圖片描述
法二:按win+Q,輸入控制面板,然後點選NVIDIA控制面板;
點選NVIDIA控制面板的幫助,點選左下角系統資訊;
點選元件:這裡就顯示了你的CUDA的資訊啦。
在這裡插入圖片描述
需注意:若控制面板顯示版本與法一 nvcc --version不一致,則以法一版本為準。因為控制面板顯示的版本應該是顯示卡支援的cuda最高版本。

3.檢視windows的cuDNN版本

首先進入資料夾C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include,找到名為cudnn.h的檔案;
在這裡插入圖片描述

然後用記事本開啟即可看到程式碼中的前部宣告部分。
在這裡插入圖片描述
即:我的cuDNN版本號為7.5.0。

需注意:cuda11.0後新的cudnn版本在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include\cudnn_version.h 檔案中定義版本號!而不是上述說的cudnn.h!

紀念一下第一次發部落格,也想在之後的科研道路上把學到的、用到的知識多多總結,多多回顧。

可能有不全面的地方,也歡迎大家多提建議,多多討論!