Bump torch

rakuri255 · Jan 3, 2025 · df48f7f · df48f7f
1 parent cc6af3b
commit df48f7f
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/install/CPU/linux_cpu.sh b/install/CPU/linux_cpu.sh
@@ -4,4 +4,4 @@ cd ..
 python3.10 -m venv .venv
 source .venv/bin/activate
 pip install -r requirements.txt
-pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2
+pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1
diff --git a/install/CPU/windows_cpu.bat b/install/CPU/windows_cpu.bat
@@ -5,5 +5,5 @@ cd ..
 py -3.10 -m venv .venv
 SET VenvPythonPath=%CD%\.venv\Scripts\python.exe
 call %VenvPythonPath% -m pip install -r requirements.txt
-call %VenvPythonPath% -m pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2
+call %VenvPythonPath% -m pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1
 endlocal
diff --git a/install/CUDA/linux_cuda_gpu.sh b/install/CUDA/linux_cuda_gpu.sh
@@ -4,4 +4,4 @@ cd ..
 python3.10 -m venv .venv
 source .venv/bin/activate
 pip install -r requirements.txt
-pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2+cu117 --index-url https://download.pytorch.org/whl/cu117
+pip install torch==2.3.1+cu118 torchvision==0.18.1+cu118 torchaudio==2.3.1+cu118 --index-url https://download.pytorch.org/whl/cu118
diff --git a/install/CUDA/windows_cuda_gpu.bat b/install/CUDA/windows_cuda_gpu.bat
@@ -5,5 +5,5 @@ cd ..
 py -3.10 -m venv .venv
 SET VenvPythonPath=%CD%\.venv\Scripts\python.exe
 call %VenvPythonPath% -m pip install -r requirements.txt
-call %VenvPythonPath% -m pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2+cu117 --index-url https://download.pytorch.org/whl/cu117
+call %VenvPythonPath% -m pip install torch==2.3.1+cu118 torchvision==0.18.1+cu118 torchaudio==2.3.1+cu118 --index-url https://download.pytorch.org/whl/cu118
 endlocal