Win10 64位下GPU版本MXNet+Tensorflow 1.3.0的安装

说明:本文记录时间为2017.7.23。在此时Tensorflow在Win下的安装要求为系统64位且Python版本为3.5(现已经支持py 3.6),MXNet在Win下只支持Python 2。
更新说明:在8月19日发布了Tensorflow1.3.0后,cudnn支持6.0而不再支持5.1,所以一定要选对版本哦。

一.安装Anaconda

由于Mxnet在win下只支持Python 2,tensorflow在win下只支持Python 3.5,所以选择Anaconda创建Python2和3共存的环境。

1. 安装Anaconda 2 作为主环境

a. 下载Anaconda
官网速度较慢,可考虑使用清华镜像 https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

b. 安装。
默认安装位置为C:\Users\Che\Anaconda2,注意需如下图所示两项都选才能将其添加到环境变量。
这里写图片描述

2. 安装Anaconda 3 环境

有两种方式安装Anaconda 3环境

  • ●conda创建
    在CMD命令行中使用如下代码创建名为mxnet的环境,使用的是Python 3.5版本。
conda create -n py3 python=3.5 anaconda

优点:简单;缺点:联网下载包,速度慢。

  • ●手动下载并安装(这种安装方式在以后的安装包的时候可能会存在问题,不建议)
    • 下载Anaconda3
    • 安装Anaconda3的安装目录必须选在C:\Users\Che\Anaconda2\envs子目录下,名称“py3”可以自己取,并加在路径的最后。安装过程中注意一定取消取消掉第一个√,避免将Anaconda3添加到系统PATH路径下。
      这里写图片描述

优点:安装时间短。

验证两种方式的Anaconda3环境创建成功与否:

a. 在CMD里面直接输入

python
exit()

会看到启动Python 2.7,然后退出。
这里写图片描述
b.使用如下代码切换到Python 3

activate py3
python
exit()

此处的”py3”即之前Python3安装目录文件夹的名字。使用activate py3命令之后,在命令行前面会出现一个[py3]标记,此时使用任何的python命令都是在Python 3.5下进行的。使用如下命令可取消激活Python 3。

deactivate py3

这里写图片描述

二.安装CUDA和cuDNN

因为我们要安装GPU版本,所以需要 CUDA 和 cuDNN 的支持。
1. 请先在 这里确认你的显卡支持 CUDA
2. 如果以上条件符合,那么可以开始从各自官网下载 CUDA 8.0 和 cuDNN V6.0的安装包

CUDA: https://developer.nvidia.com/cuda-downloads
这里写图片描述

cuDNN: https://developer.nvidia.com/rdp/cudnn-download (需要注册)

这里写图片描述
3. 安装CUDA。双击执行安装程序,安装后会自动添加系统变量。
测试安装是否成功:在CMD中输入“nvcc -V”。
这里写图片描述
4. 安装cuDNN。解压后的另一个压缩文件文件继续解压,放到任何一个目录下(此处为D:\cuda),然后把所放的那个目录添加到Path 环境变量里。
这里写图片描述
这里写图片描述
将下面这些文件复制到相应位置:
D:\cuda\bin\cudnn64_6.dll —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

D:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include

D:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

三.安装MXNet

参考网址:
https://stankirdey.com/2017/07/04/upgrading-mxnet-from-0-9-to-0-10-on-windows-10/
下载准备:
https://github.com/yajiedesign/mxnet/releases 上的两个文件,如图所示:
- VC14 – prebuildbase_win10_x64_vc14.7z
- 20170721_mxnet_x64_vc14_gpu.7z
这里写图片描述
1.将下载的两个文件解压到D:\mxnet\,将之前下载的cuDNN压缩包解压到D:\mxnet\3rdparty\cudnn

2.添加环境变量MXNET_HOME,值为D:\mxnet

3.添加环境变量Path的值:
%MXNET_HOME%\lib;%MXNET_HOME%\3rdparty\cudnn\bin;%MXNET_HOME%\3rdparty\cudart;%MXNET_HOME%\3rdparty\opencv;%MXNET_HOME%\3rdparty\vc;%MXNET_HOME%\3rdparty\gnuwin;%MXNET_HOME%\3rdparty\openblas\bin;

说明:第2,3步在原文中是使用setupenv.cmd文件来执行,但自己操作中发现Path值的路径出现多加斜杠的错误,所以这里采用了手动操作的方法。

4.添加环境变量PYTHONPATH,值为:D:\mxnet\python(这一步是因为自己后续操作发现在其他位置无法导入mxnet模块,所以这里添加python的搜索路径

5.在CMD下进入D:\mxnet\python,执行

python setup.py install

这里写图片描述
这里写图片描述
测试安装成功
这里写图片描述

到这里MXNet就安装好啦。Enjoy!

四.安装Tensorflow

Tensorflow官网:https://www.tensorflow.org/install/install_windows

●准备工作
1. 确保Python版本是3.5且系统为64位(非常重要)。
2. 确保pip版本最新,可用python -m pip install -U pip 在CMD中升级pip 。
3. 确保安装Visual Studio 2015 或者 2013 或者 2010。
4. 确保已安装cuda和cuDNN。

●安装方式:
由于我们是在py3子环境下安装了Python3,所以需先使用activate py3进入子环境。进入后,可使用官网提供的两种方式进行安装。

  • pip
pip3 install --upgrade tensorflow-gpu
  • Anaconda
pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.2.1-cp35-cp35m-win_amd64.whl

注意:

在安装Tensorflow时很可能会出现 “Cannot remove entries from nonexistent file c:\program files\anaconda3\lib\site-packages\easy-install.pth” 的报错,这是setuptools版本太低导致。
解决方法:pip install –upgrade –ignore-installed setuptools

  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值