当前位置:网站首页>[Pytorch] Install mish_cuda

[Pytorch] Install mish_cuda

2022-08-09 09:13:00 .Yunzhe.

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

版权声明
本文为[.Yunzhe.]所创,转载请带上原文链接,感谢
https://yzsam.com/2022/221/202208090903330237.html