nvidia-smi确认:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118验证:
import torch
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))RTX 3060 8G 不建议直接跑 13B 原生模型
适合:
ollama run qwen2:7b调试:
ollama ps
nvidia-smi常用参数:
ollama run qwen2:7b --num_gpu 1 --num_ctx 4096适合高级调试
./main -m qwen2-7b-q4.gguf -ngl 32 -c 4096关键参数:
-ngl:GPU 层数(8G 建议 20–32)-c:上下文长度-b:batch size常见错误:
CUDA out of memory解决:
contextllama.cpp 而不是纯 PyTorch示例(Qwen):
<|im_start|>user
你好<|im_end|>
<|im_start|>assistant| 问题 | 原因 | 解决 |
|---|---|---|
| 显存不够 | 模型太大 | 量化 |
| 很慢 | CPU 推理 | 开 GPU |
| 卡死 | context 太大 | 减小 |
| 无输出 | prompt 错 | 检查格式 |
你可以告诉我:
我可以直接给你针对性调试命令。