关于windows下Tensorflow和pytorch安装教程
一、Tensorflow安装
1、Tensorflow介绍
Tensorflow是广泛使用的实现机器学习以及其它涉及大量数学运算的算法库之一。Tensorflow由Google开发,是GitHub上最受欢迎的机器学习库之一。Google几乎在所有应用程序中都使用Tensorflow来实现机器学习。例如,如果您使用到了Google照片或Google语音搜索,那么您就间接使用了Tensorflow模型。它们在大型Google硬件集群上工作,在感知任务方面功能强大。
2、Tensorflow安装(cpu版本)
我的环境是Anaconda3.5.2m,安装的是旧版本的tensorflow
pipinstalltensorflow==1.8.0
3、新版本安装
pipinstalltensorflow pipinstall-U--ignore-installedwraptenum34simplejsonnetaddr pipinstalltensorflow
4、验证代码
importtensorflowastf hello=tf.constant('Hello,TensorFlow!') sess=tf.Session() print(sess.run(hello))
5、警告YourCPUsupportsinstructionsthatthisTensorFlowbinar......处理
#忽略警告处理方法 importos os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
6、GPU版本安装(win10+1050ti)1、下载安装CUDA(找到安装包一直下一步)
2、安装CUDNN(需要登录才能下载)
解压CUDNN安装包
打开CUDA安装位置
把CUDNN中的bin、lib、include目录中的文件复制到CUDA对应的目录中
3、安装tensirflow-gup
pipinstalltensorflow-gpu==1.8.0
二、pytorch安装
pytorch官网:https://pytorch.org/
1、pytorch介绍
pytorch是一个python优先的深度学习框架,是一个和tensorflow,Caffe,MXnet一样,非常底层的框架。
Torch自称为神经网络界的Numpy,因为他能将torch产生的tensor放在GPU中加速运算(前提是你有合适的GPU),就像Numpy会把array放在CPU中加速运算.所以神经网络的话,当然是用Torch的tensor形式数据最好。就像Tensorflow当中的tensor一样。pytorch是一个动态的建图的工具。不像Tensorflow那样,先建图,然后通过feed和run重复执行建好的图。相对来说,pytorch具有更好的灵活性。
2、安装
打开pytorch官网,然后根据情况选择安装的版本。
我的python版本是3.6.5
pipinstallhttps://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl pipinstallhttps://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
3、验证代码
importtorch print(torch.__version__)
总结
以上所述是小编给大家介绍关于windows下Tensorflow和pytorch安装教程,希望对大家有所帮助!
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。