[XingangPan/DragGAN]【教程贴】windows本地成功部署流程,常规问题解决

2024-05-10 292 views
3

git下载这些基本流程就不说了,说一下要注意的几个地方

  1. 需要安装VS环境,custom_ops.py 文件会调用,需要改一下代码配置VS路径
  2. 安装Python的时候注意要勾选debug版本
  3. 要安装cuda,版本要和torch对应,
  4. pip3 install --pre torch torchvision torchaudio --index-url https:download.pytorch.org/whl/nightly/cu121

会遇到的报错:

ImportError: DLL load failed while importing bias_act_plugin: 找不到指定的模块。 【cuda版本要一致】 LINK : fatal error LNK1104: 无法打开文件“python39.lib” ninja: build stopped: subcommand failed. 【python需要开发版,并复制lib到cuda调用文件夹下】

其他问题我比较顺利,遇到的话可以帮着一起解决,刚开始研究,有兴趣的小伙伴可以一起探讨:

IMG_7918

回答

4

cuda要是121版本,其他都OK

2

为什么不直接来个懒人包呢?谢谢大神!thanks big god!

1

cu121装不上

4

cuda要是121版本,其他都OK

不同的显卡版本不一样,N卡命令行输入 nvidia-smi 右上角查看版本

4

还有群吗