安装TensorFlow GPU版本

首先需要准备的东西:Anaconda环境

查看自己GPU型号并安装英伟达驱动程序

首先要知道自己电脑的GPU版本型号:右击我的电脑->管理->设备管理器。

比如我的时NVIDIA GeForce GTX 1060 6GB,

英伟达GeForce系列 ,型号为GTX1060 6GB运行内存,知道这个以后再去官网去看自自己相应的软件驱动

安装TensorFlow GPU版本

查询自己的GPU所对应的驱动:CUDA GPU | NVIDIA Developer 查询到我的算力时6.1

安装TensorFlow GPU版本

下载GPU驱动   Download Drivers | NVIDIA

安装TensorFlow GPU版本

点击search进入下载驱动程序

安装TensorFlow GPU版本

建议就按照软件的默认路径装,这样可能会把上一把不正确的删除掉,安装完成后的界面如下:

安装TensorFlow GPU版本

安装cudatoolkit  

安装TensorFlow GPU版本

CUDA与显卡驱动:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

TensorFlow-GPU与CUDA cudnn Python版本关系:https://tensorflow.google.cn/install/source_windows?hl=en#gpu
 

打开这个   输入命令:

  1. conda install cudatoolkit=10.1
  2. conda install cudnn=7.6
  3. pip install tensorflow-gpu==2.1.2

上一篇:解决Ubuntu重启之后NVIDIA驱动屡次失效:报错NVIDIA-SMI has failed because it couldn't communicate with the NVIDI


下一篇:anaconda 安装GPU版本的pytorch