确保你的 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 图片免费生成工具
作者:
发表回复