stylegan2 训练终极解决(包括训练环境搭建)

背景:

   需要搭建stylegan2训练环境,训练自己的MAT模型,

python环境依赖真实烦人,用docker镜像是方案。

谚词:懂原理,可灵活搭建。要符合大脑的学习规律,懂了,而且要懂原理

stylegan2官方的github网址

GitHub - NVlabs/stylegan2-ada-pytorch: StyleGAN2-ADA - Official PyTorch implementation

方法:

1、python3.7

2、python库:pip install click requests tqdm pyspng ninja imageio-ffmpeg==0.4.3

3、pytorch环境,这个最变态,经常会遇到版本问题造成的莫名其妙的事情

记住一定要用pip 安装不要用conda来安装。

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

执行训练命令:python train.py --outdir=training-runs --data=cifar10.zip --gpus=1

报错1 

ImportError: numpy.core.multiarray failed to import

 安装 numpy 1.18.0版本

报错2

ModuleNotFoundError: No module named 'typing_extensions'

执行

 pip install -U typing-extensions

到此完整解决,是简单吧,网上各种说法,你得有火眼金睛和明白的大脑。

你可能感兴趣的:(stylegan2,模型训练)