码迷,mamicode.com
首页 > Windows程序 > 详细

win10安装tensorflow-gpu

时间:2019-08-04 23:58:24      阅读:411      评论:0      收藏:0      [点我收藏+]

标签:png   解压   地址   info   https   put   pkg   http   channels   

安装cuda

先安装VS,然后根据自己的版本安装CUDA、

安装完后,打开cmd命令行输入nvcc -V,检测是否安装成功

 技术图片

 

安装cuDDN

安装对应版本,解压后覆盖到CUDA的地址,默认为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

 

使用anaconda安装tensorflow-gpu

创建一个新的环境 

conda create -n env_name python=version 

激活并进入环境中 

conda activate tensorflow

更换清华源(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 --set show_channel_urls yes

安装

conda install tensorflow-gpu

测试

 技术图片

 

win10安装tensorflow-gpu

标签:png   解压   地址   info   https   put   pkg   http   channels   

原文地址:https://www.cnblogs.com/flyuz/p/11300542.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!