安装mish-cuda(速度比pytorch官方要快)
-
安装环境:
- python3.8.10
- pytorch 1.8.1
- cuda 11.1
安装命令
git clone https://github.com/JunnYu/mish-cuda.git cd mish-cuda python setup.py build install
测试
python test/perftest.py -b
使用
import torch from mish_cuda import MishCuda mish = MishCuda() x = torch.randn(3,5) mish(x)
Tips
- https://github.com/WongKinYiu/ScaledYOLOv4 中会使用到该mish-cuda
- 当你觉得github下载速度过慢的时候可以考虑使用镜像, 只需要将 github.com 替换成 hub.fastgit.org . 如:https://hub.fastgit.org/junnyu/mish-cuda.git