在开始之前,请先确定安装好显卡驱动、CUDA、cuDNN,可以参考我的另一篇博客

  1. 安装Numpy
    1
    pip install numpy
  2. 安装Tensorflow-gpu
    1
    pip install tensorflow-gpu