add prepare.sh script

Author: TsingZ0

README.md

@@ -301,8 +301,10 @@ Install [CUDA](https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.htm
 
 Install [conda latest](https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh) and activate conda. 
 
+For additional configurations, refer to the `prepare.sh` script.  
+
 ```bash
-conda env create -f env_cuda_latest.yaml # You may need to downgrade the torch using pip to match the CUDA version
+conda env create -f env_cuda_latest.yaml  # Downgrade torch via pip if needed to match the CUDA version
 ```
 
 ## How to start simulating (examples for FedAvg)

docs/quickstart.html

@@ -182,6 +182,7 @@ <h2>Installation</h2>
                     <li><p>Then install <a href="https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html">CUDA</a> following NVIDIA's instructions.</p></li>
                     <li><p>Install the latest version of <a href="https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh">conda</a> and activate conda.</p></li>
                     <li><p>Install the required Python dependencies using <code>env_cuda_latest.yaml</code> and conda:</p></li>
+                    <p>For additional configurations, refer to the <code>prepare.sh</code> script.</p>
                     <pre>conda env create -f env_cuda_latest.yaml # You may need to downgrade the torch using pip to match the CUDA version.</pre>
                 </ol>
             </section>