diff --git a/README.md b/README.md index 98018a3..592e7eb 100644 --- a/README.md +++ b/README.md @@ -7,7 +7,7 @@ ## Installation ``` -$ conda create --name simclr python=3.7 --file requirements.txt +$ conda env create --name simclr --file env.yml $ conda activate simclr $ python run.py ``` diff --git a/env.yml b/env.yml new file mode 100644 index 0000000..bb6cde3 --- /dev/null +++ b/env.yml @@ -0,0 +1,21 @@ +name: simclr +channels: + - pytorch + - anaconda + - conda-forge + - defaults +dependencies: + - cudatoolkit=10.1 + - numpy=1.18.1 + - opencv=3.4.2 + - pillow=7.0 + - pip=20.0 + - python=3.7.6 + - pytorch=1.4.0 + - torchvision=0.5 + - tensorboard=2.1 + - matplotlib + - scikit-learn + - pyyaml=5.3.1 + - nvidia-apex=0.1 +