Skip to content

Commit 88b317f

Browse files
authored
Merge pull request #57 from onnx/gs/onnx-1.2
fix onnx version in readme
2 parents c118261 + ee36261 commit 88b317f

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

README.md

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -13,16 +13,18 @@ Tf2onnx is in its early development. Mileage will vary since TensorFlow supports
1313
Basic net and conv nets should work. A list of models that pass tests can be found [here](tests/run_pretrained_models.yaml)
1414

1515
# Installation
16-
Install dependencies:
16+
If you don't have tensorflow installed already, install the desired tensorflow build, for example:
1717
```
18-
pip install onnx==1.2
1918
pip install tensorflow
19+
or
20+
pip install tensorflow-gpu
2021
```
21-
If you want to run unit tests against the Caffe2 onnx backend, build and install Caffe2 from: ```
22-
https://github.com/pytorch/pytorch```
23-
We tested with tensorflow 1.5,1.6,1.7,1.8 and anaconda 3.5,3.6.
2422

25-
Once dependencies are installed, from the tf2onnx root folder call:
23+
If you want to run unit tests against the Caffe2 onnx backend, build and install Caffe2 following the instructions here: ```
24+
https://caffe2.ai/```
25+
We tested with tensorflow 1.5,1.6,1.7,1.8 and anaconda 3.5,3.6.
26+
27+
Once dependencies are installed, from the tensorflow-onnx folder call:
2628
```
2729
python setup.py install
2830
```
@@ -31,10 +33,13 @@ or
3133
python setup.py develop
3234
```
3335

34-
To create a wheel for distribution:
36+
To create a distribution:
3537
```
36-
python setup.py bdist_wheel
38+
python setup.py sdist
3739
```
40+
41+
tensorflow-onnx requires onnx-1.2.2 or better and will install/upgrade onnx if needed.
42+
3843
# Usage
3944
```
4045
python -m tf2onnx.convert

0 commit comments

Comments
 (0)