Open JiaXiao243 opened 5 years ago
在window下执行python ./train.py --batch_size=1 -- --use_gpu False --parallel True --train_crop_size=769 --total_step=50 --norm_type=gn --init_weights_path=./deeplabv3plus_gn_init --save_weights_path=model --dataset_path=./data/cityscape,使用CPU对deeplabV3进行训练,报错信息如下: 由于window下使用GPU进行训练,提示显存不足,故无法做对比分析。
@JiaXiao243 windows的GPU显存多大?
window下有1个显卡,显存8G, 内存最多再分给GPU8G, 用于GPU的使用,总共有16G可用;
在window下执行python ./train.py --batch_size=1 -- --use_gpu False --parallel True --train_crop_size=769 --total_step=50 --norm_type=gn --init_weights_path=./deeplabv3plus_gn_init --save_weights_path=model --dataset_path=./data/cityscape,使用CPU对deeplabV3进行训练,报错信息如下: 由于window下使用GPU进行训练,提示显存不足,故无法做对比分析。