From 4de60ad90710d1c4e731438d0106faf67faa471c Mon Sep 17 00:00:00 2001 From: Saarthak Kapse <44542181+saarthak-kapse@users.noreply.github.com> Date: Wed, 29 Jan 2025 15:20:58 -0500 Subject: [PATCH] Update README.md Updating installation instructions --- README.md | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 6de0595..61e416e 100644 --- a/README.md +++ b/README.md @@ -26,14 +26,16 @@ State Space Models (SSMs) with selective scan (Mamba) have been adapted into eff ## Environment setup -- Python 3.11.9 +- Python 3.10.13 ``` - conda create -n fastvim -c conda-forge python=3.11.9 gcc=11.4 gxx=11.4 - conda activate fastvim +conda create -n fastvim python=3.10.13 +conda activate fastvim ``` -- torch 2.1.1 + cu12.1, and other requirements. +- torch 2.1.1 + cu11.8, and other requirements. ``` +conda install --channel nvidia/label/cuda-11.8.0 +pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118 pip install -r requirements.txt ```