win10 安装Mish-CUDA 遇到问题解决

在安装的时候遇到了很多问题。

首先就是ninja报错Command '['ninja', '-v']' returned non-zero exit status 1.

然后是gcc问题

解决方法:我怀疑是torch版本不对,我使用了pytorch1.7.0和1.8.0;

所以开始更新cuda+cudnn=11.3;pytorch=1.10

此时安装成功,因为跳过了ninja的代码,

然后对mish-cuda进行,python setup.py build 和 install就可以了

ps:我修改了mish-cuda的一些代码,如果有小伙伴不能安装的话可以私我,或者在我发布的资源里找。

猜你喜欢

转载自blog.csdn.net/bieliwuguiqi/article/details/124844778
今日推荐