From 73a944b677831281218a84329cc03d98e446b6fd Mon Sep 17 00:00:00 2001 From: welix Date: Tue, 22 Oct 2024 18:22:59 +0900 Subject: [PATCH] Update README.md: Fix cuda version --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 33fcfdf..6b1fbac 100755 --- a/README.md +++ b/README.md @@ -22,9 +22,9 @@ Install ffmpeg first. If you are an Ubuntu user, run: apt install ffmpeg ``` Then, install pytorch, torchvision, torchaudio, and torchtext based on your GPU environments. -Note that the inference API is available for CPU environments. We tested the codes on Python 3.9 and CUDA 11.7: +Note that the inference API is available for CPU environments. We tested the codes on Python 3.9 and CUDA 11.8: ``` -pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 torchtext==0.15.1 +pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 torchtext==0.16.0 --index-url https://download.pytorch.org/whl/cu118 ``` Finally, run to install dependency libraries: ```