深度学习开发环境搭建 (2)

Windows不自带python,目前tensorflow在Win中支持的python版本为3.5、3.6。我们使用3.6版本。直接下载python3.6安装就好了。下载地址:点这里。安装也很简单,一直next就好了。

添加python的环境变量:

“我的电脑”-“属性”-“高级系统设置”-“环境变量”,在“系统变量”中选中“Path”,点“编辑”,加上python的路径,例如“C:\Python36”。这样就可以在命令提示符中使用python了。打开命令提示符输入

python

出现下面提示说明已经安装成功:

Python 3.6.7 (v3.6.7:6ec5cf24b7, Oct 20 2018, 03:02:14)

[GCC 4.2.1 Compatible Apple LLVM 6.0 (clang-600.0.57)] on darwin

Type "help", "copyright", "credits" or "license" for more information.

>>>

2.安装virtualenv

添加pip的环境变量:

环境变量“Path”中加上python\Script的路径,例如“C:\Python36\Script”。这样就可以在命令提示符中使用pip了。

在命令提示符输入:

pip install --upgrade virtualenv

安装完成,建立一个全新的 virtualenv 环境,例如我们想建立一个叫AI的开发环境,路径为C:Users\xxx\,那么我们输入:

virtualenv --no-site-packages C:Users\xxx\AI

安装完成。使用和上文Mac中类似,不再赘述。

3.安装常用的第三方库

和上文Mac中类似,不再赘述。只需注意tensorflow选择win版支持gpu的版本。

pip install https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl

4.安装CUDA和cuDNN

因为在Windows中我们安装的是支持gpu的tensorflow,因此比Mac中多两个安装步骤。CUDA是由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。cuDNN是NVIDIA专门针对深度神经网络(Deep Neural Networks)中的基础操作而设计基于GPU的加速库。下面我们来安装它们。

首先确保电脑安装好了Nvidia显卡驱动,打开控制面板---NVIDIA控制面板---帮助---系统信息---组件,如果支持CUDA会有NVCUDA.DLL 以及支持的CUDA版本。

根据查询到的支持的CUDA版本到这里查询匹配的tensorflow和cuDNN版本。

注意一定要版本匹配!

注意一定要版本匹配!

注意一定要版本匹配!

根据查询好的版本下载正确的CUDA(下载地址:点这里)和cuDNN(下载地址:点这里)。

CUDA下载好是exe文件,直接双击安装。安装好路径像下面这样"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0"。

cuDNN下载好是一个压缩包,里面有3个文件夹。一个include,一个lib64,还有一个bin。把它们复制到上面CUDA的安装文件夹("C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0")中。

配置环境变量,在环境变量“Path”中加上这三个路径:

a."C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

b."C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

c."C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

安装完成。

5.测试运行

import tensorflow as tf

import numpy as np

hello=tf.constant('hhh')

sess=tf.Session()

print (sess.run(hello))

如果没有报错说明安装成功了,如果报错请仔细检查版本,python版本+tensorflow版本+CUDA版本+cuDNN版本。

 

最后预祝大家安装顺利!

深度学习开发环境搭建

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/zzyxww.html