硬件要求:基本需求是一台配备独立显卡(GPU)的计算机,显存最好大于6GB。推荐显卡类型为NVIDIA系列显卡、例如GTX-1080Ti、RTX-2080、RTX-2090、RTX-3060等。AMD显卡也可以,但安装时步骤会略有差别。macOS系列计算机不带GPU,运行时需要使用到苹果的专属芯片M1/M2代替GPU,虽然苹果已经专门针对大火的Stable Diffusion进行了优化,但运行时间依然可能是GPU的10倍以上。
软件要求:系统首选Windows系统或Linux系统(Ubuntu/RedHat/Centos),由于macOS不带GPU,所以不推荐使用macOS;除此之外,源代码运行是基于Python的,所以也需要用到Python以及相关的软件包,例如pytorch、cuda等。这些软件要求已经写到自动安装的脚本中,在安装时一键运行即可,用户唯一要确认的就是自己对应的操作系统。
推荐软硬件配置如表3-1所示。
表3-1 Stable Diffusion推荐软硬件配置