1.1 windows安装TensorFlow-gpu
① 准备好NVIDIA的显卡,下载安装CUDA;
网址:https://developer.nvidia.com/cuda-downloads
② 安装好之后把CUDA安装目录下的bin和lib\x64添加到path环境变量中;
③ 下载并安装cuDNN;
网址:https://developer.nvidia.com/rdp/cudnn-download
④ 解压压缩包,把压缩包中bin,include,lib中的文件分别拷贝到NVIDIA GPU Computing Toolkit\CUDA\ v9.0目录下对应目录中;
⑤ 把NVIDIA GPU Computing Toolkit\CUDA\ v9.0\extras\CUPTI\libx64\cupti64_90.dll拷贝到NVIDIA GPU Computing Toolkit\CUDA\ v9.0\bin;
⑥ 安装GPU版TensorFlow:conda install tensorflow-gpu
注意:cuDNN和CUDA的版本必须匹配
1.2 Linux安装TensorFlow-gpu
1.2.1 CUDA的安装
1、安装驱动
驱动安装参考