jedibobo的博客

Vitis-AI在生成量化模型报错 Traceback (most recent call last): File "generate_model.py", line 191, in <module> run_main() File "generate_model.py", line 185, in run_main quantize(args.build_dir,args.quant_mode,args.batchsize) File "generate_model

故事背景 VItis-AI更新了2.0，增加了对30系显卡的支持，但是VCK5000-es1卡却并不在VItis-AI2.0的支持列表中，这一度让我非常恼火。因为自己买的就是30系显卡。 问题分析 Vitis-1.4.1分支在build docker镜像时，使用的是CUDA10作为基础镜像来编译的。并且使用Pytorch1.4.0版本。而Pytorch有一个特点，就是对于30系显卡，如果Pytorch的whl编译版本不支持cu11x版本的话，那么训练框架是无法跑起来的。那么就可以利用这个特性，升级Torch