当前位置：网站首页>【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

边栏推荐

猜你喜欢

随机推荐