Win10系统下的tensorflow基于GPU的安装

2019-03-15  本文已影响0人  爬的慢的蜗牛

Win10系统下的tensorflow基于GPU的安装

1.安装Anaconda: 首先从Anaconda官网下载Anconda软件,很好的科学计算包,包含python, numpy, pandas等工具包。通常安装python3.5以上的

2.基于gpu安装:接下来下载加速GPU计算软件:从NVIDIA公司下载CUDA安装包并按提示安装;再下载cuDNN软件,注意cuDNN版本要与CUDA相配套,另外下载前需要注册并填写问卷。一般下载CUDA8.0版本以上注意安装之前要安装了visual studio

        安装cuda

cuDNN

                                安装cuda

                            自定义高级

                            勾选所有

        一路通过即可,较为easy! 但是注意的是记住你的安装路径,因为后面要用, 默认的路径可能为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA ;;添加环境变量:一般来说,如果你安装过程不搞什么特殊操作,环境变量会自己自动添加。否则可以自己添加

        安装cuDNN

            接下来解压cuDNN

            这样CUDA Toolkit 9.0 和 cuDnn 7.0就已经安装了,下面要进行环境变量的配置。

            配置环境变量

            将下面四个路径加入到环境变量中,注意要换成自己的安装路径。

            C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

            C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

            C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64

            C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp

3. 安装tensorflow: 用CMD命令打开命令编辑器,用Pip进行tensorflow安装,由于从国外网站下载较慢,并且经常出现超时错误,因此推荐用国内镜像进行下载,如清华镜像,阿里镜像豆瓣等。

        在命令中输入pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

或者 python -m pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

如果不行按命令先更新pip:

        python -m pip install --upgrade pip  -ihttps://pypi.tuna.tsinghua.edu.cn/simple

4. 测试

    4.1.查看是否使用GPU

```

    import tensorflow as tf

    tf.test.gpu_device_name()

```

    4.2.查看在使用哪个GPU

```

    from tensorflow.python.client import device_libdevice_lib.list_local_devices()

```

文末:

    pycharm与Anaconda连接

        Anaconda 所创建的任意python环境其实都是可以被pycharm利用的,在pycharm里面新建工程,点击file,下拉菜单选择setting,然后在里面选择project,下拉里面选择project interpretor,可以直接将解释器更换为Anaconda下的Python解释器,即可使用Anaconda中的科学计算包

欢迎转载,转载请注明地址

参照博客:

    https://zhuanlan.zhihu.com/p/54186298

    https://www.cnblogs.com/guoyaohua/p/9265268.html

    https://blog.csdn.net/Cs_hnu_scw/article/details/79695347

     https://blog.csdn.net/lambert310/article/details/52412059

上一篇下一篇

猜你喜欢

热点阅读