Skip to content

Commit c6b8259

Browse files
authored
Update README.md
Add blurb about python environment
1 parent f313497 commit c6b8259

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

README.md

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,19 @@ The `tf_efficientnet` and `tflite_(se)mnasnet` models require an equivalent for
137137

138138
## Script Usage
139139

140+
## Environment
141+
142+
All development and testing has been done in Conda Python 3 environments on Linux x86-64 systems, specifically Python 3.6.x and 3.7.x. Little to no care has been taken to be Python 2.x friendly and I don't plan to support it. If you run into any challenges running on Windows, or other OS, I'm definitely open to looking into those issues so long as it's in a reproducible (read Conda) environment.
143+
144+
PyTorch versions 1.0 and 1.1 have been tested with this code.
145+
146+
I've tried to keep the dependencies minimal, the setup is as per the PyTorch default install instructions for Conda:
147+
```
148+
conda create -n torch-env
149+
conda activate torch-env
150+
conda install -c pytorch pytorch torchvision cudatoolkit=10.0
151+
```
152+
140153
### Training
141154

142155
The variety of training args is large and not all combinations of options (or even options) have been fully tested. For the training dataset folder, specify the folder to the base that contains a `train` and `validation` folder.

0 commit comments

Comments
 (0)