diff --git a/README.md b/README.md index a1999f8..fc44ac8 100644 --- a/README.md +++ b/README.md @@ -21,9 +21,15 @@ For any question just drop me an [email](mailto:manuele.rusci@unibo.it). ## Getting Started ### Prerequisites -- The code is tested with PyTorch 0.4.1 and Python 3.5 +- The code is tested with PyTorch 1.13.1 and Python 3.7 - Tensorflow package is needed to load pretrained tensorflow model weights +Run the following commands to make a conda environment, +```bash +conda create -n mixed_precision python=3.7 +pip install -r requirements.txt +``` + ### Setup Set the correct dataset paths inside `data.py` . As an example: ``` diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..0c808ee --- /dev/null +++ b/requirements.txt @@ -0,0 +1,8 @@ +bokeh==2.4.3 +numpy==1.21.6 +pandas==1.3.5 +scipy==1.7.3 +tensorflow==2.11.0 +torch==1.13.1+cu117 +torchvision==0.14.1+cu117 +-f https://download.pytorch.org/whl/torch_stable.html \ No newline at end of file