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