安装以下包需要提前安装好anaconda
pytorch2onnx实战
pytorch简介: Pytorch是一个开源的Python机器学习库,由Facebook人工智能研究院维护,是目前主流的深度学习框架之一。基础学习可参考Pytorch中文文档&教程。
pytorch安装: 建议以pytorch官网的安装命令为准(亲身体验)。按照自己的配置勾选即可,建议使用pip安装,然后在anaconda prompt输入官网给出的命令行:
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(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
问题描述:“pip is configured with location that require TLS/SSL, however the ssl module in Python is not available”
去网站:https://slproweb.com/products/Win32OpenSSL.html
下载第一个(3M).exe文件安装就可以了。
问题描述:“Cannot open D:\Program Files\Anaconda3\envs\tensorflow\Scripts\pip-script.py”
这个是pip的问题,需要重新安装pip,输入以下命令行:
python -m ensurepip
easy_install pip
问题描述:安装过程中可能会出现"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: DLL load failed: 找不到指定的模块。"一般是安装的tensorflow版本太高或者太低的问题,重新安装版本成功import。
安装指定版本:
pip install tensorflow==1.15.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
最后记录几个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 环境名
有问题欢迎在评论区留言,同时本人水平有限,有错误希望大家指正。