admin管理员组文章数量:1626225
让我们直接切入正题~
目录
安装TensorFlow
国内镜像网站
安装CUDA
安装 cuDNN
测试
安装TensorFlow
TensorFlow由Google公司打包成一个 pip 安装包,可以用正常安装包的方式安装 TensorFlow ,即进入命令行执行下面这一条简单的语句
正常安装:
pip install --upgrade tensorflow # CPU版本
pip install --upgrade tensorflow-gpu # GPU版本
但此方法是在国外网站上下载包,所以下载速度慢且容易失败。
下面我们来看多快好省的安装方法吧~(ps:如果能够翻墙的话可以忽略这里)
轻松解决你的烦恼
快速安装(用国内开源镜像安装):
python2
pip install -i https://pypi.tuna.tsinghua.edu/simple/ --upgrade tensorflow # CPU版本
pip install -i https://pypi.tuna.tsinghua.edu/simple/ --upgrade tensorflow-gpu # GPU版本
python3
pip3 install -i https://pypi.tuna.tsinghua.edu/simple/ --upgrade tensorflow # CPU版本
pip3 install -i https://pypi.tuna.tsinghua.edu/simple/ --upgrade tensorflow-gpu # GPU版本
可以将上面的镜像网站地址更换成其他镜像地址
国内镜像网站
阿里云
http://mirrors.aliyun/pypi/simple/中国科技大学
https://pypi.mirrors.ustc.edu/simple/豆瓣(douban)
http://pypi.douban/simple/清华大学
https://pypi.tuna.tsinghua.edu/simple/中国科学技术大学
http://pypi.mirrors.ustc.edu/simple/
安装GPU版本的同志朝这看~
安装好TensorFlow后我们在python中输入
import tensorflow as tf
GPU版本的会提示没有找到 CUDA 和 cuDNN
安装CUDA
前往官网上下载对应的CUDA Toolkit,可参考
博主的tensorflow-gpu版本是1.13.1,所以对应下载CUDA 10.0
下载好之后直接双击安装行了,安装时间可能会有点长,请耐心等待
安装 cuDNN
前往官网上下载对应的cuDNN
下载之后是个压缩包,解压然后将bin路径添加到系统变量Path中去。
好了,现在我们的TensorFlow算是安装成功了,那来体验一下吧
测试
import tensorflow as tf
hello = tf.constant('hello,tf')
sess = tf.Session()
print(sess.run(hello))
然后你就能看见你的显卡信息了,还有输出(啊,要暴露我的渣渣显卡了<捂脸>)
参考:
1、解决国内安装tensorflow, opencv等安装不成功或下载太慢问题
2、【TensorFlow】Windows10 64 位下安装 TensorFlow - 官方原生支持
版权声明:本文标题:Windows安装TensorFlow教程(国内源安装附上各大镜像网站网址) 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/dianzi/1728942391a1180909.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论