window10安装运行tensorFlow-gpu

自己目前做视频识别需要tensorflow-gpu加速,走了很多坑重装好几次系统,终于实现,现记录如下:pycharm+tensorflow+1050TI+Python3.6=

目前网上复制严重,好多没有实践可行性。主要问题如下:使用Anconda情况下cuda不用另外安装;cuda不需要额外配置,cudnn也不需要配置。GPU版本查看使用https://www.techpowerup.com/gpuz/

第一步 安装anconda

下载anconda Anconda下载网址

进入cmd设置国内镜像

修改镜像源

https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

conda config --set show_channel_urls yes

创建环境

conda create -n 虚拟环境名称 python=3.7

激活环境

activate 虚拟环境名称

退出环境 deactivate

删除环境 conda env remove -n 虚拟环境名称

依赖的三方包

conda install tensorflow

conda install tensorflow-gpu

第二步 安装cuda cudnn

调用GPU,现在conda命令不太好用,pip好用,但关联功能没有conda省事。

使用conda命令安装 

conda install cudatookit 会自动安装cudnn

第三步 验证cudn,cudnn ,GPU 是否可用

切换到python命令行:运行以下命令

from tensorflow.python.client import device_lib as _device_lib

local_device_protos = _device_lib.list_local_devices()

第四步 安装PyCharm 配置

File->Default Setting->Project Interpreter->conda Environment

在add Local中添加本地虚拟环境。

注意问题 

conda install tensorflow-gpu 会自动关联cuda及cudnn 查看版本是否匹配

python 版本与tensorflow-gpu版本是否匹配

你可能感兴趣的:(window10安装运行tensorFlow-gpu)