pixel2style2pixe环境配置([‘ninja‘, ‘-v‘] + no module named ‘fused’)

项目场景:

复现pixel2style2pixe
GitHub:https://github.com/eladrich/pixel2style2pixel


问题描述:

1.subprocess.CalledProcessError: Command’[‘ninja‘, ‘-v‘]’returned non-zero exit status 1
以下方案可完美解决
https://wanghao.blog.csdn.net/article/details/120947224
https://blog.csdn.net/weixin_43731803/article/details/116787152

2.no module named ‘fused’
①作者提到过GCC版本太低,但更新至7.5仍出现该问题,排除选项。
②https://blog.csdn.net/c2a2o2/article/details/120596305
链接中第一个思路,将相关函数替换成pytorch的实现方法,避免编译问题,亲测无效,排除选项。
③最后发现问题是cuda版本必须大于等于10.1。


解决方案:

nvcc -V   #查看cuda的版本号

注:不要用nvidia-smi命令查看的是cuda版本号,该命令显示的是驱动的版本。

切换cuda版本操作如下:
https://blog.csdn.net/u013905398/article/details/103799621

最后根据以下链接确定torch + torchvision的版本并安装
https://blog.csdn.net/jorg_zhao/article/details/106883420
比如我的环境python3.6 + cuda10.1

pip install torch==1.7.0 torchvision==0.8.0

注:建议修改完cuda后重新安装torch,作者提到有可能默认使用之前编译的文件导致失败。

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值