1. 程式人生 > >使用nvidia-smi命令查看顯卡信息

使用nvidia-smi命令查看顯卡信息

註冊 下載安裝 file inf pycha har dnn 軟件 dia

安裝:

  1、先安裝tensorflow-gpu,需要查看對應的版本,通過pycharm運行程序時會報錯,提示需要安裝CUDA,且會指明需要版本號

  >> pip install tensorflow-gpu

  2、下載安裝CUDA軟件,選擇相應的版本及操作系統,下載並安裝;

  CUDA下載地址: https://developer.nvidia.com/cuda-downloads

  3、下載cuDNN庫,開始需要註冊等基本信息,下載並解壓,將解壓後的bin, include, lib三個文件夾拷貝到CUDA目錄下,

  我的安裝目錄為:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

  cuDNN下載地址: https://developer.nvidia.com/cudnn

  然後再使用pycharm跑程序,再第一次時很慢,後面就快了

查看:

  1、在cmd中使用命令 nvcc -V可查看cuda版本

  2、在cmd命令中nvidia-smi可查看gpu使用情況,如果不能識別命令,需要設置Path變量,我的目錄為: C:\Program Files\NVIDIA Corporation\NVSMI

下圖是跑程序時的截圖,對比cpu跑,速度快多了,因為我的筆記本是遊戲型的,cpu差一點,顯卡好一些。

技術分享圖片

使用nvidia-smi命令查看顯卡信息