安装mish-cuda（速度比pytorch官方要快）

[183****0229]

安装环境:

• 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