使用nvidia-smi命令查看顯卡信息
阿新 • • 發佈:2018-09-02
註冊 下載安裝 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命令查看顯卡信息