美文网首页
win10+anaconda3安装tensorflow-gpu和

win10+anaconda3安装tensorflow-gpu和

作者: 赖子啊 | 来源:发表于2019-05-18 04:13 被阅读0次

    刚换了硬盘,又得重新配置深度学习框架了,现在最流行的莫过于tensorflow和pytorch,我今天就尝试了安装了他们的GPU版本(虽然我的电脑垃圾配置GeForce 930M),以供之后需要的时候参考,也供其他人借鉴
    \color{red}{PS:只针对Nvidia显卡可以,具体可看这个,tensorflow有要求}
    https://developer.nvidia.com/cuda-gpus

    1、需安装的东西

    • Win10 x64位
    • Anaconda3 x64 python 3.6.5
    • CUDA Toolkt10.0
    • cuDNN 7.5 for cuda10
    • Nvidia drive 430.64
    • Visual Studio 2017 community

    2、具体实现

    2.1 windows10的安装

    在我之前写过的安双系统的一篇文章里面有提及安装win10的教程在此不作过多赘述

    2.2 Anaconda3的安装(略)

    安装过程中提示“不建议添加到环境变量中”,那个默认不勾选就好,要启动到开始里面,到Anaconda Prompt里面用管理员身份打开即可

    2.3 Visual Studio 2017 community

    现在官网上面已经提供最新的2019下载,之前的版本下载好像要订阅什么的,你们也可以下载我这里提供的链接:https://pan.baidu.com/s/1Zvtg8nHPC_oZBA9rRC_CAA
    提取码:s90s ,安装没什么难的,主要之后的显卡驱动要用到VS里面的东西所以要安装它。

    2.4 Nvidia drive 430.64的安装

    其实我感觉这个是不必须的,因为youtube上面很多教程都没有提及这个,而且我发现后来安装CUDA Toolkit的时候它也会装drive,但是我是装了,你们可以不装试试看
    去这里找你电脑的N卡对应的驱动,然后下载安装,这个过程可能需要耐心等待,其中选项都默认就行

    Nvidia驱动正在安装 下面是我安好的截图 显卡驱动安装成功 安好之后桌面上多了这样一个图标 驱动图标 之后需要重启(在我不重启的时候后面的CUDA10.0安装不成功),至此显卡驱动已安装完成!

    2.5 CUDA Toolkt10.0的安装

    去这里下载对应版本的,我是win10本地安装,如果不想安装10.1可以在Release Notes里面找10.0版本的安装

    CUDA Toolkit10.0 这个是安装中的截图: cuda10.0安装过程图 重启后cuda10安装成功 这里面都安装在C盘,默认没有改安装路径,安装成功桌面上出现这样的图标: cuda10.0安装完图标

    2.6 cuDNN 7.5 for cuda10的安装

    去这里注册下载,这个下载得先注册,选择cuDNN 7.5 for cuda10,因为我们之前装的就是CUDA10.0

    cuDNN7.5.1下载 下载完就是一个压缩包 cudnn.png ,解压出来就是一个名为“cuda”的文件夹,里面有 解压出来的cuda中的文件
    • 有些教程里面是说把这三个文件夹里面的.dll文件覆盖掉cuda安装目录下的对应文件,我分析他们之后就没有添加cuda的环境变量了,我没试过
    • 我是按照把C:\cuda\bin添加到环境变量里面了,我认为一样的,我实践证明有效

    至此要安装的东西几乎完成了

    3、 添加环境变量

    除了上面说的把C:\cuda\bin添加到环境变量里面,还要照下图的添加两个环境变量

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

    这就是刚刚CUDA的安装目录(当然上面是我自己的啦,你们要看自己的改一下)

    4、安装packages

    4.1 安装tensorflow-gpu版

    进入到Anaconda Prompt里面(因为我没有安装python,这个是anaconda自带的),因为anaconda包挺齐全的,所以直接安装

    pip3 install tensorflow-gpu
    

    我是下载下来.whl文件,然后cd到.whl目录,用

    pip install tensorflow_gpu-1.13.1-cp36-cp36m-win_amd64.whl
    

    这中间我出现过ModuleNotFoundError: No module named 'numpy.core._multiarray_umath' 错误,这个只要卸载numpy再重新安装一下就好了

    pip uninstall numpy
    pip install numpy
    

    下面我测试了一下,看看是否安装成功了


    测试tf安装成功

    4.2 安装pytorch-gpu版

    一般都是到官网去用一句话安装,如

    pytorch官网 但是这上面一般都是最新版的,点击下面的Previous versions of PyTorch,可以找到对应平台的.whl文件下载下来,和上面安装tensorflow的.whl文件一样安装pytorch。 pytorch_gpu安装成功

    至此就实现了win10+anaconda3安装tensorflow-gpu和pytorch-gpu版

    如果你觉得有用,就不要吝啬你的喜欢吧,写此教程不易,需要你们的鼓励!!

    相关文章

      网友评论

          本文标题:win10+anaconda3安装tensorflow-gpu和

          本文链接:https://www.haomeiwen.com/subject/ydhzaqtx.html