-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathinstall_dependencies.bat
29 lines (25 loc) · 1.1 KB
/
install_dependencies.bat
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
@echo off
pushd %~dp0
echo Select version to install:
echo 1. CPU (Select this if you are unsure which option to choose)
echo 2. CUDA (Faster, only for NVIDIA GPUs)
echo 3. Exit
choice /C 123 /N /M "Enter your choice (1, 2, or 3):"
if errorlevel 3 (
echo Exiting without installing dependencies
exit /b 0
) else if errorlevel 2 (
echo Uninstalling existing Pytorch
.\python-3.11.7-embed-amd64\python.exe -m pip uninstall torch torchvision
echo Installing CUDA version of PyTorch
.\python-3.11.7-embed-amd64\python.exe -m pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cu124
.\python-3.11.7-embed-amd64\python.exe -m pip install -r .\requirements.txt
) else (
echo Uninstalling existing Pytorch
.\python-3.11.7-embed-amd64\python.exe -m pip uninstall torch torchvision
echo Installing CPU version of PyTorch
.\python-3.11.7-embed-amd64\python.exe -m pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cpu
.\python-3.11.7-embed-amd64\python.exe -m pip install -r .\requirements.txt
)
echo Completed.
Pause