当前位置:网站首页>【Pytorch】安装mish_cuda

【Pytorch】安装mish_cuda

2022-08-09 09:03:00 .云哲.

git clone https://github.com/thomasbrandon/mish-cuda
cd mish-cuda
cp external/CUDAApplyUtils.cuh csrc  # cuda >= 11.0
python setup.py build install
原网站

版权声明
本文为[.云哲.]所创,转载请带上原文链接,感谢
https://blog.csdn.net/luolinll1212/article/details/113883962