diff --git a/README.md b/README.md
index fb413c5e703277494cff1126869aa5c8e552f095..0709543be42a365b982614aa4bc413e2ce7c4f60 100644
--- a/README.md
+++ b/README.md
@@ -15,7 +15,8 @@ cd path/to/code
 make
 ```
 
-Prerequisite : [conda library]( https://docs.conda.io/projects/conda/en/latest/user-guide/install/linux.html)
+Prerequisite : [conda library]( https://docs.conda.io/projects/conda/en/latest/user-guide/install/linux.html)\
+Attention: It is recommended to uninstall the pytorch library provided with the conda environment to download yourself the version most adapted to your machine from the pytorch [website](https://pytorch.org/get-started/locally/).
 
 ## Reproductibility
 
@@ -24,4 +25,3 @@ All the commands needed to reproduce the paper results are written in the `Exper
 C. Robardet, A. Batel, M. Plantevit, I. Benouaret
 >LIRIS laboratory, UMR5205, Univ Lyon, CNRS, UBL,LIRIS
 >EPITA Research Laboratory (LRE)
-## Reference