Navigation

    Gpushare.com

    • Register
    • Login
    • Search
    • Popular
    • Categories
    • Recent
    • Tags

    安装mish-cuda(速度比pytorch官方要快)

    技术交流
    1
    1
    617
    Loading More Posts
    • Oldest to Newest
    • Newest to Oldest
    • Most Votes
    Reply
    • Reply as topic
    Log in to reply
    This topic has been deleted. Only users with topic management privileges can see it.
    • 183****0229
      183****0229 last edited by

      安装环境:

      • 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
      1 Reply Last reply Reply Quote 2
      • First post
        Last post