win10 安装Tensorflow-gpu版本

第一阶段:安装CUDA和cudnn

  1. 根据电脑配置去NVIDIA官网下载相应的CUDN版本,我下载的是CUDA9.0
  2. 根据CUDA版本在NVIDIA官网下载相应的cudnn版本,此处下载的是cudnn v7.5.0
  3. 执行CUDA的exe文件开始安装
  4. 安装完之后,解压cudnn压缩包,将解压之后的文件夹中的文件复制到CUDA的相应文件夹中

第二阶段:使用anaconda安装tensorflow-gpu

  1. 下载Anaconda,我下载的是Anaconda3-2019.03-Windows-x86_64.exe(内置python版本是3.7)
  2. 安装Anaconda,安装完之后检查Anaconda是否成功安装:conda --version
  3. 检测目前安装了哪些环境:conda info --envs(目前只显示了基本环境)
  4. 设置tensorflow环境并安装tensorflow所要用的python版本:conda create --name tensorflow python=3.6(python3.6.8),亲测自带的python3.7不行
  5. 查看环境conda info --envs,其tensorflow环境已经成功添加
  6. 切换到tensorflow环境:activate tensorflow(退出当前环境:deactivate)
  7. 查看tensorflow环境下的python版本:python --version
  8. 在tensorflow环境下安装GPU版本:pip install tensorflow-gpu==1.12.0(亲测安装1.13.1不成功)

第三阶段:测试tensorflow-gpu是否安装成功

  1. 进入python命令行:python
  2. import tensorflow as tf
  3. 如果步骤2没报错,就差不多了,具体测试见下图
    在这里插入图片描述
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值