深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm

本文介绍深度学习环境配置,仅供参考。深度学习需要用到Python,本人安装Anaconda,再安装TensorFlow,然后配置到PyCharm中。

配置环境:

CPU:Intel i5-8250U

显卡:NVIDIA GeForce MX110 2G独显

内存:128G固态硬盘+1T机械硬盘,小米国民本

系统:win10家庭中文版

安装包准备:

Visual Studio Community 2015、CUDA9.2、cuDNN v7.4.1、Anaconda3-5.2.0-windows-x86_64、 tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl 、pycharm-community-2018.3

1、CUDA9.2

我的显卡是显示是CUDA 9.2.130 driver,需要下载CUDA9.2,再此之前需要安装Visual Studio Community 2015,然后下载CUDA9.2,下载地址:https://developer.nvidia.com/cuda-92-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork

建议在线安装,有两个安装包,其中一个是补丁。如果不确定显卡的版本,可以在显卡系统信息如下图所示查询。

《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》

安装时选择自定义安装,只选择CUDA,其他组件不安装,不要勾选Visual Studio Integration,否则可能会安装失败。把补丁安装一下。

2、cuDNN v7.4.1

下载地址:https://developer.nvidia.com/rdp/cudnn-download

下载cuDNN v7.4.1,对应的是CUDA9.2版本。官网下载需要注册账号,也可以自行从网上其他地方找。下载后无需安装,直接解压得到三个文件夹。把这三个文件夹里的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2 安装目录下相同名字的文件夹中,如下图。

《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》

验证CUDA安装成功:在运行里面输入“cmd” ,然后用命令“nvcc -V”进行测试,安装成功会输出cuda版本。

《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》

3、Anaconda3

下载地址:https://repo.anaconda.com/archive/

由于目前TensorFlow不支持Python3.7,所以需要安装Python3.6,选择Anaconda3-5.2.0-windows-x86_64,不要选择最新的Anaconda 5.3.1版本。安装完成后点击Anaconda Prompt,输入Python查看版本3.6.5,如下

《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》

4、在Anaconda中安装TensorFlow 1.8

下载地址:https://github.com/fo40225/tensorflow-windows-wheel/tree/master/1.8.0/py36/GPU/cuda92cudnn71sse2

目前tensorflow官方版本还不支持cuda9.2,有大神已经解决了这个问题,需要从git上将tensorflow1.8.0下载到电脑上,将下载的文件tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl放到 c:\users\jing位置 ,否则安装时可能找不到。打开Anaconda Prompt,进入Anaconda命令行管理界面:

(1)conda create –name tf python=3.6 #创建tf环境

(2)conda activate tf #激活tf环境

(3)pip install tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl #安装TensorFlow

安装完成后,测试如下:

activate tf

python

输入以下代码:

import tensorflow as tf

hello = tf.constant(‘Hello, TensorFlow!’)

sess = tf.Session()

print(sess.run(hello))

没有报错说明配置好了,如下图

《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》

5、在PyCharm中配置Anaconda和TensorFlow

下载免费的PyCharm社区版pycharm-community-2018.3,专业版需要付费,下载地址:https://www.jetbrains.com/pycharm/download/#section=windows

安装PyCharm后,需要设置,然后配置Anaconda和TensorFlow。打开PyCharm,点击Settings,进入Project Interpreter,添加Anaconda即可,如下图第二张所示,若Package中没有显示tensorflow-gpu等包,需要点击右侧的绿色按钮,如第三张图片的红色框所示,就会出现你所需要的。点击OK配置成功。最后测试是否配置好Anaconda和TensorFlow。

《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》
《深度学习配置:win10+NVIDIA显卡+Anaconda+TensorFlow+PyCharm》

到此深度学习环境配置成功。如有不足之处,还请指正。

    原文作者:well
    原文地址: https://zhuanlan.zhihu.com/p/51137303
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞