git clone https://github.com/thomasbrandon/mish-cuda
cd mish-cuda
cp external/CUDAApplyUtils.cuh csrc # cuda >= 11.0
python setup.py build install
【Pytorch】安装mish_cuda
最新推荐文章于 2022-05-26 21:54:46 发布
git clone https://github.com/thomasbrandon/mish-cuda
cd mish-cuda
cp external/CUDAApplyUtils.cuh csrc # cuda >= 11.0
python setup.py build install