如何使用a100去搭建这个项目的环境，我看他的tensorflow代码对应的cuda版本很低。

[pyh007]

如何使用a100去搭建这个项目的环境，我看他的tensorflow代码对应的cuda版本很低。

[Luh1124]

搭建不了，可以先用低版本机器处理好数据；算法运行方面可以使用高版本cuda

[erpapa]

可以试试我的分支：https://github.com/erpapa/FFHQ-UV，将dnnlib替换为pytorch版本的stylegan2，经测试可以正常生成。

[erpapa]

可以试试我的分支：https://github.com/erpapa/FFHQ-UV，将dnnlib替换为pytorch版本的stylegan2，经测试可以正常生成。

步骤如下：

1. 下载stylegan2_pytorch项目：https://github.com/adriansahlman/stylegan2_pytorch
2. 查看requirements.txt，最好是做下修改指定版本，然后再安装依赖
3. 将FFHQ-UV目录下的checkpoints/stylegan_model/stylegan2-ffhq-config-f.pkl模型拷贝到stylegan2_pytorch项目里
4. 在stylegan2_pytorch目录下，转换ffhq的tf模型：python run_convert_from_tf.py --input stylegan2-ffhq-config-f.pkl --output stylegan2-ffhq-config-f-G.pth stylegan2-ffhq-config-f-D.pth stylegan2-ffhq-config-f-Gs.pth
5. 将stylegan2-ffhq-config-f-Gs.pth模型拷贝到FFHQ-UV库目录里的checkpoints/stylegan_model目录下
6. 回到FFHQ-UV项目，执行正常的生成流程