要求

如果您有 CUDA 显卡，请遵循NVlabs/stylegan3的要求。

通常的安装步骤涉及以下命令，它们应该设置正确的 CUDA 版本和所有 python 包

conda env create -f environment.yml
conda activate stylegan3

然后安装附加要求

pip install -r requirements

否则（对于带有 Silicon Mac M1/M2 的 MacOS 上的 GPU 加速，或仅 CPU）请尝试以下操作：

cat environment.yml | \
  grep -v -E 'nvidia|cuda' > environment-no-nvidia.yml && \
    conda env create -f environment-no-nvidia.yml
conda activate stylegan3

# On MacOS
export PYTORCH_ENABLE_MPS_FALLBACK=1

在 Docker 中运行 Gradio 可视化工具

提供的 docker 镜像基于 NGC PyTorch 存储库。要在 Docker 中快速试用可视化工具，请运行以下命令：

docker build . -t draggan:latest  
docker run -p 7860: 7860 -v "$PWD":/workspace/src -it draggan:latest bash  
cd src && python visualizer_drag_gradio.py --listen

现在您可以从 Gradio 打开共享链接（在终端控制台中打印）。
请注意，Docker 镜像大约需要 25GB 的磁盘空间！





			隐藏内容
		

					
本内容需权限查看

			

			  
			


			


				
    
				
  • 普通用户: 30积分
  • VIP会员: 免费
  • 永久会员: 免费
    • 				


			


			
已有1人解锁查看
声明：任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。