a312863063 / generators-with-stylegan2

Here is a series of face generators based on StyleGAN2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

提给想用代码但是在苦苦配置环境的linux用户。(环境清单)(踩坑集锦)

Harley-ZP opened this issue · comments

基于这个stylegan2,

linux环境有以下要求:
1.gcc版本不能高于9
2.gcc版本要至少符合cpp11特性
3.cuda只能是10.0
4.python37及以上应该是不支持的

分享我最后跑通的环境:
tf-gpu 1.14.0
gcc 7.3.1
python 3.6
cuda 10.0
cudnn 也要匹配 cuda

其实也可以考虑去dockerhub找找玩这个模型的人分享的容器镜像😀。

你好,请问下您知道 该模型的docker容器镜像 名字是什么吗?

你好,问下你的cuda版本为10.0,那么cudnn版本是多少呢?

事实上这个环境配置和硬件也有关系,如果你是30系的显卡,就无法支持低版本cuda。
如果采用高版本cuda,又无法与 tf 兼容。