Pytorch2onnx2Tensorflow的环境准备win10

安装以下包需要提前安装好anaconda
pytorch2onnx实战

pytorch安装

pytorch简介: Pytorch是一个开源的Python机器学习库,由Facebook人工智能研究院维护,是目前主流的深度学习框架之一。基础学习可参考Pytorch中文文档&教程
pytorch安装: 建议以pytorch官网的安装命令为准(亲身体验)。按照自己的配置勾选即可,建议使用pip安装,然后在anaconda prompt输入官网给出的命令行:
安装pytorch

tensorflow安装

tensorflow简介: tensorflow是一种采用数据流图,用于数值计算的开源软件库,也是目前主流的深度学习框架之一,由google团队维护。tensor表示传递的数据是张量(多维数组),flow表示对计算图进行运算,即用图表示计算任务,并且图必须在会话中执行

tensorflow安装: 建议在anaconda中新建一个名为“tensorflow”的虚拟环境,该环境自带了能够运行python的基本包,这么做的原因是在新建的环境中操作不会影响外部环境,想怎么搞怎么搞,当然直接在外部环境安装tensorflow也是OK的。
新建环境:

conda create --name tensorflow python=3.6

指定的python版本是新环境中的python版本,与外部环境无瓜。如果要安装3.7版本,有的conda版本可能不支持,可以先查看可安装的python版本:

conda search  --full --name python

激活环境:

activate tensorflow

命令行前出现“(tensorflow)”就说明当前是在tensorflow环境下。
如果工作完成可退出环境:

deactivate

以上完成的是新建环境过程,我们只是把这个环境命名为tensorflow,里面并没有安装tensorflow,接下来展示如何在新环境中安装tensorflow:

pip install tensorflow

或者:

pip install --upgrade --ignore-installed tensorflow

等待安装,安装完可输入pip list查看tensorflow版本。然后验证tensorflow是否安装成功,当前环境下输入python,会跳出python版本信息,并进入python编辑,输入以下命令:

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

输出b'hello!就说明安装成功了!
参考博客:手把手教你如何安装Tensorflow(Windows和Linux两种版本)

onnx和onnx_tf安装

onnx简介: onnx(Open Neural Network Exchange)是一个开放神经网络交换格式,是一种通用深度学习框架,我们正是通过onnx将pytorch模型转为tensorflow模型。
onnx安装:

pip install onnx

onnx_tf安装:

pip install onnx_tf

或者去https://github.com/onnx/onnx-tensorflow下载,完成后进入onnx-tensorflow目录下输入命令行:pip install -e

可能遇到的问题及解决方案

SSL模块不可用

问题描述:“pip is configured with location that require TLS/SSL, however the ssl module in Python is not available”
SSL problem
去网站:https://slproweb.com/products/Win32OpenSSL.html
SSL下载
下载第一个(3M).exe文件安装就可以了。

Cannot open …pip-script.py

问题描述:“Cannot open D:\Program Files\Anaconda3\envs\tensorflow\Scripts\pip-script.py”
这个是pip的问题,需要重新安装pip,输入以下命令行:

python -m ensurepip
easy_install pip

ReadTimeoutEorror

问题描述:安装过程中可能会出现"ReadTimeoutEorror"之类的错误,是因为pip默认的资源库是国外的,没有科学上网的话下载速度很慢,超过默认的时间就会报错,这类问题有两种解决方法:
如果安装的包比较小: 可以设置默认的超时时间,默认是15,我们可以设为100,比如我们更新pip到最新版本(只有1.4M):

python -m pip install --default-timeout=100 --upgrade pip

如果安装的包比较大: 可以将资源库换成国内的镜像源:

pip install 包名 -i 镜像源

国内镜像源主要有:
https://pypi.tuna.tsinghua.edu.cn/simple(清华)
https://pypi.douban.com/simple(豆瓣)
http://mirrors.aliyun.com/pypi/simple/(阿里云)
比如我们安装tensorflow(几百M):

pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple

如果不想每次安装都加个-i 镜像源,可以点击Pycharm中的File—settings—project interpreter—“+”号—manage repositories,将默认源换成清华源或其他源。
这里有更详细的国内镜像源使用方法:https://www.cnblogs.com/microman/p/6107879.html

import tensorflow报错"ImportError"

问题描述:import tensorflow的时候报错:"ImportError: DLL load failed: 找不到指定的模块。"一般是安装的tensorflow版本太高或者太低的问题,重新安装版本成功import。
安装指定版本:

pip install tensorflow==1.15.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

conda常用指令

最后记录几个conda常用指令:
安装指定版本的包:conda install package=version
指定包的来源:conda install --channel 镜像源
恢复默认源:conda config --remove-key channels
检测目前安装了哪些环境变量:conda info --envs
查看当前有哪些可使用的python版本:conda search --full --name python
安装新环境:conda create --name 环境名

有问题欢迎在评论区留言,同时本人水平有限,有错误希望大家指正。

  • 2
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值