本文进入热榜收到了不少关注，所以将本文的代码放在了GitHub上，jupyter的，有需要的自取。

torch.nn.DataParallel()这个主要适用于单机多卡。个人一般比较喜欢在程序开始前，import包之后使用os.environ['CUDA_VISIBLE_DEVICES']来优先设定好GPU。例如要使用物理上第0,3号GPU只要在程序中设定如下：...

链接：https://www.zhihu.com/question/452749603/answer/1826252757

Apache TVM 是一个相对较新的 Apache 项目，以深度学习模型推理的性能大幅改进为目标。它属于一种叫做模型编译器(model compilers) 的新技术: 它以高级框架(如 PyTorch 或 TensorFlow)中编写的模型作为输入，生成一个为...

地址：https://zhuanlan.zhihu.com/p/363319763

说一下是YOLOv5的第五个版本，不是YOLO的第五个版本！是YOLOv5又又改进了！

不久之前，PyTorch 官方博客发布 1.8 版本，此版本由 1.7 发布以来的 3000 多个 commit 组成，重要更新内容包括在编译、代码优化和科学计算前端 API 方面的优化和改进，以及通过 pytorch.org 提供的二进制文件来支持 AMD ROC...

来源丨https://zhuanlan.zhihu.com/p/337850513

来源丨https://zhuanlan.zhihu.com/p/361101354

up在试了网上很多教程之后，一次次的都错，真的是安装了无数遍GPU版本的pytorch，使用清华源镜像但是有个巨坑，查了很多博客，终于有了点头绪顺利解决安装问题速度飞快。...