如何解决"Torch is not able to use GPU"错误?



我想在Windows10中建立一个稳定的扩散环境。我下载了推荐的显卡驱动程序版本和 cuda 版本,但运行 webui-user-bat 仍然生成错误:Torch 无法使用 GPU。 在此处输入图像描述 在此处输入图像描述

我怀疑图形驱动程序版本与 cuda 版本不匹配,但我尝试了许多版本,但没有一个有效。有人说火炬的版本不匹配。我目前的火炬是官方网站上的最新版本。

在根文件夹中执行 webui.sh shell 脚本,然后重试运行 webui-user.bat。

我一直在研究一个已知问题,我认为它归结为用户需要在根文件夹中执行脚本 webui.sh(使用 bash 或类似方式执行),它应该安装 ROCM。我不明白这些细节,但它允许用户使用 AMD GPU。以下是一些更多信息:

https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/9970

让我简化一下:

当我开始时,我遇到了这个问题,但现在我知道如何处理它,这也是你可以做到的。

仅适用于 CPU:

如果您没有 GPU 并希望在 CPU 上使用它,请按照以下步骤操作:

  • 导航到您的文件夹。
  • 找到"webui-user.bat"文件。
  • 左键单击它并选择"编辑"。
  • 查找显示"set commandline_args="的行,并向其添加"--skip-torch-cuda-test
  • "(应看起来像set commandline_args= --skip-torch-cuda-test)。

如果您有 GPU 并想要使用它:

您所需要的只是一块至少具有 2 GB 内存的 NVIDIA 显卡。有人说你需要一张超过 4 GB 的卡,但我有一张叫 MX300 的卡,它有 2 GB 内存,我仍然可以使用它,没有任何问题,以实现稳定的扩散。唯一的缺点是生成图片需要 2 到 4 分钟,具体取决于几个因素。顺便说一句,也有一种方法可以在AMD GPU上运行它,但我对此知之甚少。

在 GPU 上使用它的步骤:

  • 首先,下载为GPU使用而制作的稳定扩散的特殊版本。你可以在这里找到它:https://github.com/automatic1111/stable-diffusion-webui/wiki/install-and-run-on-nvidia-gpus它还告诉你如何下载和使用它,但它是过时的信息。
  • 打开下载的文件夹并找到"run.bat"文件。
  • 双击它,如果它只是火炬,它将下载启用了 cuda 的火炬,而不是转到步骤 3 顺便说一句,我更喜欢你执行第 3 步,即使它使用 cuda 下载。默认情况下,它会下载旧版本的火炬。

第 3 步:

  • 在下载"torch"之前,请关闭您正在处理的文件。
  • 访问此网站: https://pytorch.org/get-started/previous-versions/
  • 在该网站上找到适合您设备的"火炬"版本。
  • 打开命令提示符 (cmd)。
  • 复制"run.bat"文件的路径。
  • 将路径粘贴到命令提示符中。
  • 添加"-m"和您从网站获得的"torch"命令。它应该看起来像这样:"pathtothefile -m pip install torch==1.13.0+cu116...">
  • 现在,运行该命令。

该过程完成后,只需再次运行"run.bat"文件,它将下载一些必要的库,然后您可以使用它。

注意:

拥有 GPU 但仍然无法在稳定扩散上运行它们的原因是他们的版本错误,如果您有多个 GPU 并想使用其中的特定 GPU,请转到"webui-user.bat"文件并将这一行添加到"set cuda_visible_devices=1"下方的"set commandline_args="。默认情况下,稳定扩散将单独使用最好的 GPU,但这是一个可选步骤。 而且我不认为它会是必要的,但是如果您仍然在GPU上使用它时遇到问题,则需要从计算机上的NVIDEA网站下载CUDA,这是唯一可以尝试的解决方案。如果您使用的是任何与AI相关的模型,将其安装在计算机上也是一种好方法

如上所述,在按照以下语法审核"webui-user.bat"文件后,它正在按预期工作我的应用程序。

导航到您的文件夹。 找到"webui-user.bat"文件。 左键单击它并选择"编辑"。 查找显示"set commandline_args="的行,并向其添加"--skip-torch-cuda-test"(应看起来像set commandline_args= --skip-torch-cuda-test)。

相同的另一个来源: - https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/1742

相关内容

  • 没有找到相关文章

最新更新