Skip to content

Commit 8e590a6

Browse files
committed
Update cuda version
1 parent d52f14b commit 8e590a6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Makefile

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ all: $(targets)
88
.PHONY: setup lock
99
setup:
1010
conda install pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 pytorch-cuda=12.1 -c pytorch -c nvidia
11-
conda install -c nvidia cuda-toolkit=11.6
11+
conda install -c nvidia cuda-toolkit=12.4
1212
pip install flash-attn --no-build-isolation
1313
pip install -U pip
1414
pip install -Ur requirements.txt

0 commit comments

Comments
 (0)