确保你的 GPU 支持 CUDA(NVIDIA 显卡)。安装正确的 PyTorch 版本,例如:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
检查 CUDA 版本是否兼容 nvcc --version。
确保你的 GPU 支持 CUDA(NVIDIA 显卡)。安装正确的 PyTorch 版本,例如:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
检查 CUDA 版本是否兼容 nvcc --version。
AI 生成头像 👉 AI 头像在线生成
作者:
发表回复
要发表评论,您必须先登录。