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

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容