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


安裝:

  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差一點,顯卡好一些。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM