docs: 初始装机指南
包含 6 章步骤 + 排错手册: - 01 硬件与基础检查 - 02 Windows 装 Studio 驱动 - 03 WSL2 + CUDA 12.8 - 04 Ollama 装 LLM - 05 PyTorch 验证(待补,依赖实机跑通) - 06 ComfyUI 装出图(待补,依赖实机跑通) - 90 排错手册
This commit is contained in:
@@ -0,0 +1,98 @@
|
||||
# 05 PyTorch 验证
|
||||
|
||||
## 目标
|
||||
|
||||
装 PyTorch 2.7.1 + cu128,验证能调用 5060 Ti。
|
||||
|
||||
## 装 venv
|
||||
|
||||
**WSL2 Ubuntu 终端**:
|
||||
```bash
|
||||
cd /mnt/d/llm-code
|
||||
uv venv --python 3.11 .venv
|
||||
source .venv/bin/activate
|
||||
```
|
||||
**应当看到**命令行前缀变成 `(.venv) eric@...`。
|
||||
|
||||
## 装 PyTorch(用清华源代理,国内最快)
|
||||
|
||||
```bash
|
||||
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 下载)。
|
||||
|
||||
## 验证(关键!)
|
||||
|
||||
```bash
|
||||
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
|
||||
```
|
||||
|
||||
## 跑个真测试(确认能算东西)
|
||||
|
||||
```bash
|
||||
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`
|
||||
|
||||
按顺序排查:
|
||||
1. `nvidia-smi` 在 WSL2 能不能看到卡
|
||||
2. `nvcc --version` 是不是 12.8
|
||||
3. 重新装一遍 PyTorch(可能是装时网络断了,下了不完整的 wheel)
|
||||
|
||||
### 装时报 `pip` 相关错
|
||||
|
||||
```bash
|
||||
deactivate
|
||||
uv venv --python 3.11 .venv --clear
|
||||
source .venv/bin/activate
|
||||
# 重装
|
||||
```
|
||||
|
||||
### 装完 import 报 `libcudart.so not found`
|
||||
|
||||
CUDA Toolkit 没装好,回到 [03 WSL2 + CUDA 12.8](./03-wsl2-ubuntu.md) 重新装。
|
||||
|
||||
## 下一步
|
||||
|
||||
✅ 通过后 → [06 ComfyUI 装出图](./06-comfyui-image.md)
|
||||
Reference in New Issue
Block a user