Update README.md

README.md

@@ -25,24 +25,20 @@ If you prefer to use **Conda**, you can download and install **Anaconda** or **M
 
 - **Miniforge** is a smaller, community-based installer for Conda that includes **Mamba**. Download it [here](https://github.com/conda-forge/miniforge/releases/latest).
 
-After installation, you can use `conda` or `mamba` for environment management. The commands will be the same except for replacing `conda` with `mamba`.
+After installation, you can use conda for environment management.
 
 ---
 
 ### 2. **Create a New Environment**
 
-Once you have Conda or Mamba installed, follow these steps to create a new environment and install the necessary packages.
+Once you have Conda (https://conda.io/projects/conda/en/latest/user-guide/install/index.html), follow these steps to create a new environment and install the necessary packages.
 
 #### **Step 1: Create a new environment**
 
-You can create a new environment called `lwm_env` (or any other name) with Python 3.12 or any required version:
+You can create a new environment called `lwm_env`:
 
 ```bash
-# If you're using Conda:
-conda create -n lwm_env python=3.12
-
-# If you're using Mamba:
-mamba create -n lwm_env python=3.12
+conda create -n lwm_env
 ```
 
 #### **Step 2: Activate the environment**
@@ -50,7 +46,6 @@ mamba create -n lwm_env python=3.12
 Activate the environment you just created:
 
 ```bash
-# For both Conda and Mamba:
 conda activate lwm_env
 ```
 
@@ -62,7 +57,7 @@ Install the necessary packages inside your new environment.
 
 # Install CUDA-enabled Pytorch
 ```bash
-pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
 ```
 # Install other required packages from conda-forge
 ```bash