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

边栏推荐

猜你喜欢

随机推荐