windows下安装tensorflow-GPU

一、CUDA安装

  检查电脑是否支持GPU类型。
在这里插入图片描述
  点击设备管理器---->显示示配器
在这里插入图片描述  进入官网查看显卡是否支持CUDA:https://developer.nvidia.com/cuda-gpus
在这里插入图片描述

  安装CUDA

   请看清楚您要安装的tensorflow版本对应的CUDA版本号!
   请看清楚您要安装的tensorflow版本对应的CUDA版本号!

点击链接查看:https://tensorflow.google.cn/install/source#linux
在这里插入图片描述
  在https://developer.nvidia.com/cuda-toolkit-archive
链接中下载
在这里插入图片描述

  由于官网下载速度慢,因此找到网页下载按钮。鼠标右击,复制链接地址。在迅雷中打开链接进行下载。
在这里插入图片描述
取消勾选GeForce Experience
在这里插入图片描述
如果电脑上本身就有Visual Studio Integration,要将这个取消勾选,避免冲突
在这里插入图片描述

点开Driver comonents,Display Driver这一行,前面显示的是Cuda本身包含的驱动版本是411.31
如果你电脑目前安装的驱动版本号新于Cuda本身自带的驱动版本号,那一定要把这个勾去掉。否则会安装失败(相同或者小于驱动版本号的话,就不用去取勾了)
在这里插入图片描述
剩下的直接安装即可。
  安装成功后 在 命令行输入 nvcc -V 会显示版本信息 则安装成功。
在这里插入图片描述

如果没有 查看系统环境变量 的 path 中是否有CUDA 的相关目录
在这里插入图片描述

安装CUDNN

   请看清楚您要安装的tensorflow版本对应的CUDNN版本号!
  CUDNN官网链接:https://developer.nvidia.com/rdp/cudnn-download

  选择相应的版本号:
在这里插入图片描述
解压CUDNN,将其复制到CUDA文件夹下
在这里插入图片描述

系统环境配置

添加环境路径:
  控制面板——>系统和安全——>系统——>高级系统设置
在这里插入图片描述
  点击环境变量,在系统变量中找到PATH

在这里插入图片描述
  双击打开查看,添加好以下路径:

在这里插入图片描述
在这里插入图片描述

安装tensorflow
pip install tensorflow-gpu==1.13.1

安装完成!

参考博文:https://www.cnblogs.com/Assist/p/11002756.html
     https://www.sohu.com/a/323645104_100058348

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值