win10安装tensorflow-gpu 2.0.0

win10安装tensorflow-gpu 2.0.0

  • 前言
  • 特别提醒
  • 安装
    • 1)下载并安装miniconda
    • 2)设置镜像源
    • 3)安装tensorflow
    • 4)测试
  • 感谢
  • 参考链接

前言

本篇文章主要介绍win10下tensorflow2的安装,以方便他人搭建环境,少走弯路。

由于个人需要,在电脑装了tensorflow2,之后重装系统,所以需要重新搭建环境。随着硬软件和驱动的发展,重装系统后与之前有所不同,重新搭建环境遇到了很多问题,当然,按照经验,无非就是安装anaconda、CUDA、cudnn、添加环境变量、conda install tensorflow-gpu即可,然而,现实狠狠地抽了我一个大嘴巴子,我遇见最大的难题:tensorflow可以安装,但只能安装低版本的tensorflow1,否则眼睁睁看着tensorflow2安装好,import时却显示Import Error:DLL load failed,找不到指定模块 ,我所使用的程序也都随着学习更新为tensorflow2版本,所以是能安装的不能用,能用的又不能安装,网上说的各种NVIDIA驱动重装,安装VS2015、CUDA和cudnn版本对应等方法全都试过,甚至一度想再次重装系统,历时两天(几乎崩溃,我对自己也都失去了信心,请读到这里的有缘人不要嘲笑我笨),终于搞定,最终总结了以下简单方法,无需手动安装CUDA和cudnn,至于NVIDIA,一般带N卡的驱动电脑装机时已经装好,没有的或者不小心卸载的可以去官网下载:https://www.nvidia.cn/。本次安装基于miniconda,个人也更建议使用miniconda,因为miniconda几乎只有python和conda,而anaconda则更加庞大,有很多的库,容易造成库之间的不兼容,我安装过程中tensorflow2安装后import报错一直解决不掉可能就是某些库冲突导致。

特别提醒

安装前请将之前安装的相关程序全部清除!!!
整个过程简单方便,概括为以下几步:
1)下载并安装miniconda
2)设置镜像源
3)安装tensorflow
4)测试

安装

1)下载并安装miniconda

首先去官网下载miniconda:https://conda.io/miniconda.html,找到想要下载的版本,解压后打开安装程序,除以下页面时第一个选项打勾,自动添加环境变量外,无需多余设置,一直下一步即可。(囧,这是anaconda的安装截图,由于我装miniconda时没有截图,所以借用了之前视频里的图)
win10安装tensorflow-gpu 2.0.0_第1张图片

2)设置镜像源

打开目录:C:\Users\你的计算机名称,找到名为.condarc的文件,用记事本打开,将其内容全部删除,替换为以下内容,保存。(此文件包含conda的默认源设置,更改此文件原因是conda安装一些库的网址都是国外网址,不掌握科学上网时下载速度超慢,甚至连接失败,而国内有很多不错的镜像源,使得我们下载这些库时速度非常快,我们这一步就是加入国内镜像源,使得用conda命令下载时可以调用国内镜像源下载。)

channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

原文链接:https://blog.csdn.net/leo12354/article/details/105868494

3)安装tensorflow

不做赘述,点击:开始菜单 - Anaconda Prompt(miniconda3),命令窗口打开后直接运行以下命令(本人安装了 2.0.0 版本,此处以安装 2.0.0 为例)

conda install tensorflow-gpu==2.0.0

等待安装,显示successful…即安装成功!

4)测试

按照上一步打开命令窗口,或者win + R,运行 cmd ,打开命令窗口,依次输入

#进入python环境
python		# 运行后显示python版本
import tensorflow as tf
tf.__version__		# 查看tensorflow版本
tf.test_gpu_available()		# 测试gpu是否能用,不出意外返回True

至此,tensorflow2 安装完成,开始你的表演吧!

感谢

感谢网上各位热心大佬分享经验,正是各位的优秀分享解决了我遇到的各类问题,此文也是希望能够能量传递,望有缘人在需要时能够起到一些作用,同时也感谢读者耐心看完,第一次编辑,编辑有误或者表述有问题也请烦请指出,期待我们的共同进步!

参考链接

[1]: https://blog.csdn.net/leo12354/article/details/105868494? utm_medium=distribute.pc_relevant.none-task-blog-baidujs-1
[2]: https://www.bilibili.com/video/BV1MC4y1p7H2?from=search&seid=8818474182450893236

你可能感兴趣的:(tensorflow2,python,tensorflow,深度学习,机器学习,神经网络)