代码详见我上传的资料。
经过我的测试cuda10.2下
torch1.7和torch1.5都编译失败了
只有torch1.4 cuda10版本编译通过。
编译时记得把build文件删除,不然会影响本地环境的编译。
编译就运行 python setup.py develop即可。
代码详见我上传的资料。
经过我的测试cuda10.2下
torch1.7和torch1.5都编译失败了
只有torch1.4 cuda10版本编译通过。
编译时记得把build文件删除,不然会影响本地环境的编译。
编译就运行 python setup.py develop即可。