- 博客(2)
- 收藏
- 关注
原创 Pytorch运行模型时报错总结
Default process group is not initialized 在使用分布式的文件里 import torch.distributed as dist 后加上下面的语句,或者在报错的地方前面加上这句。 dist.init_process_group('gloo', init_method='file:///tmp/somefile', rank=0, world_size=1) ...
2022-03-30 14:32:43 1203 6
原创 build vitis-ai-gpu docker镜像
由于docker hub 里只有vitis-ai的cpu版本镜像可以直接通过pull命令倒入,但是一些程序运行还是需要gpu版本,因此需要自己build vitis-ai-gpu。 官网给的代码是这样的 Use below commands to build the GPU docker: cd setup/docker ./docker_build_gpu.sh To run the GPU docker, use command: ./docker_run.sh xilinx/vitis-
2022-03-03 13:34:56 1440
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人