diff --git a/05-pytorch-verify.md b/05-pytorch-verify.md index 3f197ce..75c3d7d 100644 --- a/05-pytorch-verify.md +++ b/05-pytorch-verify.md @@ -2,7 +2,16 @@ ## 目标 -装 PyTorch 2.7.1 + cu128,验证能调用 5060 Ti。 +装 PyTorch + cu128,验证能调用 5060 Ti。 + +> **本机实测版本**(2026-06-14): +> - Python 3.11.15 +> - torch 2.12.0 +> - torchvision 0.27.0 +> - torchaudio 2.11.0 +> - triton 3.7.0 +> - CUDA Toolkit 12.8(系统层) + CUDA Runtime 13.0.96(PyTorch 自带) +> - 下载耗时:5 分钟(清华源代理) ## 装 venv @@ -31,6 +40,25 @@ uv pip install torch torchvision torchaudio \ **预计 5-15 分钟**(约 2.5 GB 下载)。 +### 实机安装日志(2026-06-14) + +``` +Resolved 33 packages in 2.11s +Prepared 33 packages in 5m 00s +Installed 33 packages in 5m 19s ++ torch==2.12.0 ++ torchaudio==2.11.0 ++ torchvision==0.27.0 ++ triton==3.7.0 ++ nvidia-cublas==13.1.1.3 ++ nvidia-cuda-runtime==13.0.96 ++ nvidia-cudnn-cu13==9.20.0.48 ++ nvidia-nccl-cu13==2.29.7 ++ ...(共 33 个包,含 NVIDIA 13.x 系列 CUDA 库) +``` + +> **注意**:装 PyTorch 2.12 时它**自动拉了 CUDA 13.x 的运行时包**(cublas/cudnn/nccl 等),但**用的是系统层的 CUDA Toolkit 12.8**(`nvcc`)来编译扩展。这两套并存,**没问题**——PyTorch 编译时认 12.8,运行用 13.0,版本对得上。 + ## 验证(关键!) ```bash @@ -42,6 +70,15 @@ python -c "import torch; print('CUDA:', torch.cuda.is_available(), '| GPU:', tor CUDA: True | GPU: NVIDIA GeForce RTX 5060 Ti ``` +### 实机输出(2026-06-14) + +``` +(.venv) eric@ERIC-GEM12:/mnt/d/llm-code$ python -c "import torch; print('CUDA:', torch.cuda.is_available(), '| GPU:', torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'NONE')" +CUDA: True | GPU: NVIDIA GeForce RTX 5060 Ti +``` + +✅ **PyTorch 链路全打通**。 + ## 跑个真测试(确认能算东西) ```bash diff --git a/README.md b/README.md index 5000913..6da895e 100644 --- a/README.md +++ b/README.md @@ -17,7 +17,7 @@ | [02 Windows 装 Studio 驱动](./02-windows-driver.md) | ✅ 已跑通 | nvidia-smi 看到 5060 Ti + 610.47 | | [03 WSL2 + CUDA 12.8](./03-wsl2-ubuntu.md) | ✅ 已跑通 | nvcc --version 显示 12.8 | | [04 Ollama 装 LLM](./04-ollama-llm.md) | ✅ 已跑通 | qwen2.5:7b 聊天正常 | -| [05 PyTorch 验证](./05-pytorch-verify.md) | ⏳ 待补 | 看到 `CUDA: True \| 5060 Ti` | +| [05 PyTorch 验证](./05-pytorch-verify.md) | ✅ 已跑通 | 看到 `CUDA: True \| 5060 Ti`(torch 2.12.0)| | [06 ComfyUI 装出图](./06-comfyui-image.md) | ⏳ 待补 | 浏览器开 8188 出图 | | [90 排错手册](./90-troubleshoot.md) | ✅ 已有 | 镜像源 / zstd / WSL2 透传失败 |