Skip to content

Commit

Permalink
Fix conda command
Browse files Browse the repository at this point in the history
  • Loading branch information
lolipopshock authored May 2, 2024
1 parent 27cd230 commit 9ee3ef0
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,9 @@ Use the following code to create the environment:

```bash
conda create -n co-llm python=3.11 -y
conda activate co-llm
conda install \
pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0
pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 \
pytorch-cuda=12.1 -c pytorch -c nvidia -y

pip install -r requirements.txt
Expand Down Expand Up @@ -158,4 +159,4 @@ If you find our work helpful, please cite it as:
journal={arXiv preprint arXiv:2403.03870},
year={2024}
}
```
```

0 comments on commit 9ee3ef0

Please sign in to comment.