Windows配置tensorflow-gpu环境:最新,最快速,容错率最高的方法,没有之一。

21 篇文章 33 订阅
7 篇文章 1 订阅

本方法优点

  • 不需要逐个自己安装cuda, cudnn甚至vs2015等坑爹玩意
  • 不会直接在自己电脑上装cuda,防止各种路径混乱之类。
  • 可以多个cuda兼容操作
  • 可以完美的在不影响已有tf1.0系列的情况下,完美使用tf2.0
    对于绝大部分读者,第一条已经是足够的理由了,毕竟现在大部分的博客攻略还是大费周章的。

准备工作

以我电脑为例,但本方法绝对不限于此种配置

系统Windows 10, 显卡1080Ti

前期准备

  1. 从英伟达官网更新显卡驱动至最新版本,以前受某些博客影响,担心驱动过新无法向前兼容。今天破釜沉舟之下发现, 最新驱动仍可以兼容Cuda9.0,因此不会导致以前的cuda无法使用。相反,最新驱动可以让你支持最新的cuda
  2. 官网下载Anaconda3最新版本。根据本人经验,只要你想做python开发且是新手,迟早要下的。

开工

  1. 打开 Anaconda Prompt。 (下完anaconda后自动会出现,开始菜单的Anaconda3可以找到)输入以下命令行创建新环境:
    conda create -n tf2 python=3.6这段代码的意思就是新创建一个名为tf2的python3.6环境。这是Anaconda广为使用的原因之一,很容易切换多个python版本。
    紧接着运行activate tf2,进入tf2环境。(成功的话会发现命令行前面括号里从’base’变成了’tf2’)
  2. 下载tensorflow-gpu。这一步以tensorflow2.0为例, 继续在Anaconda窗口运行pip install tensorflow-gpu==2.0.0-alpha0。很快安装成功(速度太慢的话自行百度切换pip镜像源
  3. 依次运行conda install -c anaconda cudatoolkitconda install cudatoolkit cudnn。本来这两个东西是要自己去官网下载安装还要配各种path变量之类的。幸而,conda帮你完成了。 有几大好处:
  • 极其省事
  • 不需要配置变量名,防止出错
  • 只下载了cudatoolkit,没有下载一整个,空间节约了三倍以上
  • cuda下载在当前的虚拟环境(如这里的tf2)而非在C盘整个系统环境中。因此可以同时兼容多个cuda版本,环境间互不干扰

完工

测试一下:可以直接在Anaconda Prompt里输入pythonJ进入python环境
然后运行

import tensorflow as tf
x = tf.constant([1])

如果没有报任何错误,那么已经成功配置!
重复上述操作,可以创建许多不同tensorf版本不同cuda版本的环境,来支持不同的版本的代码运行!

后话

下午本着尝试tf2.0的心态,惊喜的发现了这一操作。相比去年安装cuda9实在是轻松了太多。然而现在还没有详细叙述这一方法的博文,因此本着分享的心态,以这篇博文志之,希望对需要帮助的有所裨益。所有问题都可以直接评论讨论,也可以邮箱联系lint17@fudan.edu.cn

Tips

tensorflow2.0的API修改有点多,目前来看文档太少,大家可以不急着升级。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

B417科研笔记

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值