我目前正在开发一个网络应用程序,允许用户上传肖像图像的集合(总共10-15个),并且作为回报,生成以各种场景中的对象为特征或具有特定属性的一系列输出图像,例如以宇航员的身份出现。应用程序的主要功能集中在利用预在一些实施例中,图像处理器可以使用经训练的机器学习模型来处理输入图像并基于用户定义的属性或上下文生成期望的输出。
从我的研究来看,pixel 2style 2 pixel似乎是一个合适的预训练模型。但是,我没有NVIDIA GPU,需要只使用我的CPU在本地运行应用程序(我知道这并不理想)。我尝试让pixel2style2pixel
在CPU上工作,但我遇到了问题,它无法运行。
下面是我的Python项目的requirements.txt
文件:
torch==1.9.0+cpu
torchvision==0.10.0+cpu
Pillow==9.3
-f https://download.pytorch.org/whl/torch_stable.html
fastai>=2.3.1
ipywidgets
voila
但我不断收到与CUDA相关的错误,例如:
No CUDA runtime is found, using CUDA_HOME='C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA\v11.6'
有几件事需要注意:
- 我试过安装各种只支持CPU的PyTorch版本,但它仍然不起作用。
- 我的操作系统是Windows。
- 作为一名具有Java软件工程背景的开发人员,我最近开始了探索机器学习应用程序开发领域的旅程。尽管我在这个特定领域的经验有限,但我在软件工程方面的基础知识使我能够快速获得新技能,并通过参考相关文档和投入足够的时间来理解特定主题来有效地连接点。
我正在寻找如何实现这一点的建议,或者是否有任何其他工具/技术堆栈,我可以用来创建这个Web应用程序,而不需要NVIDIA GPU。任何帮助都是感激!
先谢谢你了!
1条答案
按热度按时间1zmg4dgp1#
你的pytorch没问题。你看到的是从这里传来的:
from torch.utils.cpp_extension import load
熔凝活性剂
_find_cuda_home()输出警告“未找到CUDA运行时,...”
但您可以忽略此警告。
更重要的是,代码试图编译一些cuda文件,如果没有cuda和nvidia gpu,这将无法工作。我检查了'load'函数调用,结果发现这些文件中有2个:
熔凝活性剂
upfirdn2d.py
原来stylegan2的主人有这些操作的cpu代码
熔凝活性剂
upfirdn2d.py
如果您删除或注解掉torch.utils.cpp_extension.load行和其他对jit编译模块的引用,并从stylegan2master复制cpu代码,就可以跳过这些。
虽然notebook有它的用途,但在IDE中调试这类问题更好,所以我建议设置它,并逐行处理这类问题,否则可能很难看到确切的问题类型。