Windows10 64位安装TensorFlow-GPU

时间:2021-03-29 22:51:43

TensorFlow有GPU版和CPU版。

GPU版需要CUDA和cuDNN支持,到链接:https://developer.nvidia.com/cuda-gpus 确认自己的显卡是否支持CUDA。本人采用pip方式安装GPU版本的TensorFlow。

安装Python:

  官网下载Python:https://www.python.org/downloads/windows/,默认设置安装。

安装TensorFlow:

  #GPU版本:

  pip3 install --upgrade tensorflow-gpu

  #CPU版本:

  pip3 install --upgrade tensorflow

  安装过程由于网络原因会出现错误,一般为红色。多试几次就可以安装成功。

安装CUDA:

  根据官方安装:https://www.tensorflow.org/install/install_windows提示,CUDA版本为8.0,cuDNN版本为v6或者v6.1,CUDA的下载链接:https://developer.nvidia.com/cuda-toolkit-archive,选择相应的版本。

安装cuDNN:

  官网链接:https://developer.nvidia.com/rdp/cudnn-download下载对应CUDA版本的cuDNN,解压添加路径(例如:D:\STORE\cuda\bin)到path变量

若官网维护可到百度云下载,链接:http://pan.baidu.com/s/1o7NM3Z8 密码:40qx

  Windows10 64位安装TensorFlow-GPU

  

测试结果:

  Windows10 64位安装TensorFlow-GPU

。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。

GPU版本的是安装成功了,仔细看看终端里面的文字: Ignoring visible gpu device (device: 0, name: GeForce 610M, pci bus id: 0000:01:00.0) with Cuda compute capability 2.1. The minimum required Cuda capability is 3.0.大致意思说,忽略了cuda的计算能力是2.1版本的GPU,需要的cuda的计算能力的最低版本为3.0.所以我的显卡太烂了 ,用不了。随后附上显卡支持的CUDA版本链接: