确保你的 GPU 支持 CUDA（NVIDIA 显卡）。安装正确的 PyTorch 版本，例如：
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
检查 CUDA 版本是否兼容 nvcc --version。
喜欢这款？立即生成同款美图 👇
AI 创作中心
×
📸
点击上传 / 拖拽图片
数值越小越像原图，数值越大 AI 发挥空间越大。
AI 小可正在努力创作，请耐心等待… 0%
发表回复
要发表评论，您必须先登录。