Files

T

包含 6 章步骤 + 排错手册：
- 01 硬件与基础检查
- 02 Windows 装 Studio 驱动
- 03 WSL2 + CUDA 12.8
- 04 Ollama 装 LLM
- 05 PyTorch 验证（待补，依赖实机跑通）
- 06 ComfyUI 装出图（待补，依赖实机跑通）
- 90 排错手册

2026-06-14 12:08:38 +08:00

2.3 KiB

Raw Blame History

05 PyTorch 验证

目标

装 PyTorch 2.7.1 + cu128，验证能调用 5060 Ti。

装 venv

WSL2 Ubuntu 终端：

cd /mnt/d/llm-code
uv venv --python 3.11 .venv
source .venv/bin/activate

应当看到命令行前缀变成 (.venv) eric@...。

装 PyTorch（用清华源代理，国内最快）

uv pip install torch torchvision torchaudio \
  --index-url https://download.pytorch.org/whl/cu128 \
  --extra-index-url https://pypi.tuna.tsinghua.edu.cn/simple

参数	用途
`--index-url`	强制走 PyTorch 官方源（拉 cu128 专用 wheel）
`--extra-index-url`	补充源，其他依赖走清华（`pypi.tuna.tsinghua.edu.cn`）

不要同时加 --index-url 和 -i，uv 会报"重复"。

预计 5-15 分钟（约 2.5 GB 下载）。

验证（关键！）

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

跑个真测试（确认能算东西）

python -c "
import torch
x = torch.randn(1000, 1000, device='cuda')
y = torch.randn(1000, 1000, device='cuda')
z = x @ y
print('矩阵乘法 OK, 形状:', z.shape, '设备:', z.device)
print('显存占用:', torch.cuda.memory_allocated()/1e9, 'GB')
"

应输出：

矩阵乘法 OK, 形状: torch.Size([1000, 1000]) 设备: cuda:0
显存占用: 8.0... GB

镜像版本必须匹配

torch	torchvision	torchaudio
2.7.0	0.22.0	2.7.0
2.7.1	0.22.1	2.7.1
2.8.0	0.23.0	2.8.0

不要混搭（如 torch 2.7.1 + torchvision 0.22.0，会出兼容警告）。

常见问题

输出 `CUDA: False`

按顺序排查：

nvidia-smi 在 WSL2 能不能看到卡
nvcc --version 是不是 12.8
重新装一遍 PyTorch（可能是装时网络断了，下了不完整的 wheel）

装时报 `pip` 相关错

deactivate
uv venv --python 3.11 .venv --clear
source .venv/bin/activate
# 重装

装完 import 报 `libcudart.so not found`

CUDA Toolkit 没装好，回到 03 WSL2 + CUDA 12.8 重新装。

下一步

✅ 通过后 → 06 ComfyUI 装出图

2.3 KiB Raw Blame History Unescape Escape