首先在github上找到大神的源码并下载下来。
https://github.com/alexandrosstergiou/SoftPool
softpool介绍不是本文重点,具体可参考:
https://blog.csdn.net/WZZ18191171661/article/details/113048529
需要的依赖环境:
pytorch >=1.4
CUDA (我是在11.0版本编译成功)
python 3.8(3.7也可以)
这里需要将CUDA_PATH改成v11.0
保证上面的环境安装成功。
将源码文件夹下载并解压,我将名字修改为SoftPool,并将文件夹复制到用conda管理的环境中:
E:\Anaconda3\envs\album\Lib\site-packages
进入到pytorch文件夹中 运行 python setup.py install
如下表示已经安装成功
测试是否安装成功,在测试的时候发现导入softpool_cuda 还是报错(心情有点崩溃,配置环境有时候就是这样,一个错误可以卡很久)
在百度上找啊找,原来问题如此简单
https://blog.csdn.net/xc_zhou/article/details/104093110
就是在PATHEXT中加上**.DLL**
好了,在测试一下:
OK啦