美文网首页
01.基础安装

01.基础安装

作者: 乌和兔 | 来源:发表于2020-04-15 12:16 被阅读0次

    1.前言:

    1. Mac不支持显卡cuda;
    2. Docker搭建Anaconda环境并安装Tensorflow,只有Linux版;
    3. Windows虚拟机不支持访问gpu和cuda;

    因此win10系统安装Tensorflow:

    1. 直接搭建Anaconda环境并安装Tensorflow;
    2. 安装双系统,切换到Linux;

    2.具体安装:

    2.1 准备工作

    确实自己要安装gpu版本还是cpu版本;
    cpu版本安装较简单,安装按照官网教程即可。

    2.2 确定gpu型号

    然后用谷歌搜索“你的GPU型号+SPECIFICATION”,在官网查它是否支持CUDA,且算力是否合适

    2.3 安装CUDA

    直接官网下载,但需要版本与所需要的tensorflow版本对应

    2.4 安装cuDNN

    下载CUDNN后,复制粘贴cuDNN里面的下面三个文件到CUDA的相应同名文件

    • cudnn\cuda\bin => CUDA\v10.0\bin
    • cudnn\cuda\include => CUDA\v10.0\include
    • cudnn\lib\x64 => CUDA\v10.0\lib\x64

    2.5 用ANACONDA安装TENSORFLOW

    1.在ANACONDA里创建名为TENSORFLOW的环境

    你可以叫他任何名字,这里我叫这个环境为TENSORFLOW

    conda create -n tensorflow pip python=3.6

    2.发出以下命令以激活 conda 环境:

    activate tensorflow

    3.conda 环境中安装 TensorFlow

    pip install --ignore-installed --upgrade tensorflow_gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

    4.测试是否安装成功

    相关文章

      网友评论

          本文标题:01.基础安装

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