安装TensorFlow GPU版本

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

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

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

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

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

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

下载GPU驱动   Download Drivers | NVIDIA

点击search进入下载驱动程序

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

安装cudatoolkit  

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

https://tensorflow.google.cn/install/source_windows?hl=en#gpu

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. pip3 install tensorflow_gpu-2.2.0

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值