36 lines
1.1 KiB
Batchfile
36 lines
1.1 KiB
Batchfile
@echo off
|
||
chcp 65001 >nul
|
||
echo 安装PyTorch GPU版本（通过官方源）
|
||
echo ===================================
|
||
echo.
|
||
|
||
echo 请选择CUDA版本:
|
||
echo 1. CUDA 12.8 (适用于较新的NVIDIA GPU)
|
||
echo 2. CUDA 12.6 (适用于较旧的NVIDIA GPU)
|
||
echo 3. CUDA 11.8 (适用于较旧的NVIDIA GPU)
|
||
echo.
|
||
|
||
set /p choice=请输入选项 (1/2):
|
||
|
||
if "%choice%"=="1" (
|
||
echo 正在安装PyTorch CUDA 12.8版本...
|
||
uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
|
||
)
|
||
else if "%choice%"=="2" (
|
||
echo 正在安装PyTorch CUDA 12.6版本...
|
||
uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
|
||
)
|
||
else if "%choice%"=="3" (
|
||
echo 正在安装PyTorch CUDA 11.8版本...
|
||
uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
|
||
) else (
|
||
echo 无效的选项！
|
||
goto end
|
||
)
|
||
|
||
echo.
|
||
echo 验证PyTorch GPU支持状态...
|
||
python -c "import torch; print('CUDA是否可用:', torch.cuda.is_available()); print('PyTorch版本:', torch.__version__); print('CUDA版本:', torch.version.cuda if torch.cuda.is_available() else '无')"
|
||
|
||
:end
|
||
pause |