关于windows下Tensorflow和pytorch安装教程

作者:巽逸 时间:2023-06-09 07:35:54 

一、Tensorflow安装

1、Tensorflow介绍

Tensorflow是广泛使用的实现机器学习以及其它涉及大量数学运算的算法库之一。Tensorflow由Google开发,是GitHub上最受欢迎的机器学习库之一。Google几乎在所有应用程序中都使用Tensorflow来实现机器学习。 例如,如果您使用到了Google照片或Google语音搜索,那么您就间接使用了Tensorflow模型。它们在大型Google硬件集群上工作,在感知任务方面功能强大。

2、Tensorflow安装(cpu版本)

我的环境是Anaconda3.5.2m,安装的是旧版本的tensorflow


pip install tensorflow==1.8.0

3、新版本安装


pip install tensorflow
pip install -U --ignore-installed wrapt enum34 simplejson netaddr
pip install tensorflow

4、验证代码


import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

5、警告Your CPU supports instructions that this TensorFlow binar......处理


#忽略警告处理方法
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

6、GPU版本安装(win10+1050ti)1、下载安装CUDA(找到安装包一直下一步)

关于windows下Tensorflow和pytorch安装教程

2、安装CUDNN(需要登录才能下载)

关于windows下Tensorflow和pytorch安装教程

解压CUDNN安装包

关于windows下Tensorflow和pytorch安装教程

打开CUDA安装位置

关于windows下Tensorflow和pytorch安装教程

把CUDNN中的bin、lib、include目录中的文件复制到CUDA对应的目录中

3、安装tensirflow-gup


pip install tensorflow-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官网 ,然后根据情况选择安装的版本。

关于windows下Tensorflow和pytorch安装教程

我的python版本是3.6.5


pip install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip install https://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

3、验证代码


import torch
print(torch.__version__)

总结

以上所述是小编给大家介绍关于windows下Tensorflow和pytorch安装教程,希望对大家有所帮助!

来源:https://www.cnblogs.com/zhangb8042/p/11212319.html

标签:Tensorflow,pytorch,安装
0
投稿

猜你喜欢

  • python实现网页自动签到功能

    2024-01-03 00:13:14
  • pytorch: tensor类型的构建与相互转换实例

    2023-06-14 09:22:57
  • 2009年五一劳动节logo欣赏

    2009-05-02 18:26:00
  • 最强Python可视化绘图库Plotly详解用法

    2022-03-08 11:01:07
  • 详解python列表生成式和列表生成式器区别

    2021-09-13 16:15:14
  • Python pomegranate库实现基于贝叶斯网络拼写检查器

    2021-06-17 19:27:51
  • [整理版]防止Access数据库被下载的9种方法

    2007-08-10 09:31:00
  • 火遍全网的Python二次元特效轻松掌握

    2022-01-29 22:01:32
  • MySQL中易被我们忽略的细节

    2024-01-21 09:54:48
  • js select option对象小结

    2024-04-19 09:58:40
  • php swoft框架实例用法

    2024-05-11 09:45:02
  • python K近邻算法的kd树实现

    2022-01-09 19:05:43
  • Python基于滑动平均思想实现缺失数据填充的方法

    2021-10-20 12:36:02
  • MySQL explain根据查询计划去优化SQL语句

    2024-01-26 16:59:26
  • PHP接口多继承及tarits实现多继承效果的方法

    2023-11-21 04:23:33
  • MySQL长字符截断的实现示例

    2024-01-25 23:55:01
  • 抽取oracle数据到mysql数据库的实现过程

    2024-01-14 03:07:12
  • pytorch实现好莱坞明星识别的示例代码

    2022-11-28 14:26:47
  • python实现俄罗斯方块小游戏

    2023-02-17 15:11:12
  • Web设计的成就感

    2009-09-04 19:02:00
  • asp之家 网络编程 m.aspxhome.com