wuvin
Always take risks!
Toggle navigation
wuvin
主页
实验室的搬砖生活
机器学习
公开的学术内容
公开的其他内容
About Me
归档
标签
友情链接
ZYQN
ihopenot
enigma_aw
hzwer
杨宗翰
StyleGAN2 复现踩坑记
2021-06-21 22:22:39
2685
0
1
wuvin
* 为什么 NVIDIA 的一些 submodule 非要C++现编译啊！给个只用 torch的版本不行吗！！！整整折腾了我六个小时。 * 发现原来出问题的地方有点多 * GCC 不能用 `0.6.5` 需要用 `0.7.5` * pytorch 需要 `1.8.1` 以上(最后用了 `1.9.0`)，否则 pytorch 无法正确调用 ninja `1.10.0` * CUDA 版本需要 11.1 及以上，以及必须使用手动安装的 cudatoolkit（即安装到 `/usr/local/cuda`），conda 自动安装以及驱动安装时安装的那个不行。请用 torch.version.cuda 查询cuda实际使用的版本。 * 似乎还需要python 3.8以上，3.7.2的环境满足上述要求的情况下也无法运行。 * 如果只需要复现一次，可用 colab 环境，安装 ninja 后即可复现。 ----------- * 更新一下：必须使用`1.8.1`，`1.9.0`能Inference 不能 Train ------------ * 更新一下：改回1.8.1会出现模块无法编译 TAT ------------ * 我最后放弃conda了！docker 大法好！！
上一篇:
DGP 代码细节
下一篇:
20210621 NeRF专题笔记
0
赞
2685 人读过
新浪微博
微信
腾讯微博
QQ空间
人人网
提交评论
立即登录
, 发表评论.
没有帐号?
立即注册
1
条评论
More...
文档导航
wuvin
没有帐号? 立即注册