作者可以出一个支持pytorch1.7的吗

[nostayup]

买了3090，但是目前机器只允许pytorch1.7，哭……

[lucasjinreal]

为什么只允许1.7, 这个周末我适配一下看看

[nostayup]

配

就是编译那一大段输出中会有报一些错，dcn_v2_cpu.o这个文件生成失败。我摘出来一部分：
2_cpu.cpp:82:9: error: ‘THFloatBlas_gemm’ was not declared in this scope

ninja: build stopped: subcommand failed.

File "/home/user/anaconda3/envs/bgmatting/lib/python3.7/site-packages/torch/ut ils/cpp_extension.py", line 1238, in _write_ninja_file_and_compile_objects
error_prefix='Error compiling objects for extension')
File "/home/user/anaconda3/envs/bgmatting/lib/python3.7/site-packages/torch/ut ils/cpp_extension.py", line 1538, in _run_ninja_build
raise RuntimeError(message) from e

[nostayup]

为什么只允许1.7, 这个周末我适配一下看看

看到一篇帖子是解决这个问题的，但是我照着做了之后惨遭失败，不知道是否对你有帮助
https://blog.csdn.net/qq_36810544/article/details/110197821

[nostayup]

CharlesShang/DCNv2#92
这里有一个解决方案，亲测可用

[caiduoduo12138]

我还是编译不过去
Python3.7
cuda11+torch1.7
还是显示不支持compute_x86
你怎么搞得
你Torch怎么装的？conda还是pip 给终端代码，我再试试

[nostayup]

我还是编译不过去
Python3.7
cuda11+torch1.7
还是显示不支持compute_x86
你怎么搞得
你Torch怎么装的？conda还是pip 给终端代码，我再试试

就是正常的pytorch上面装cuda11.0的那个conda命令行，cuda就装了11.1，然后就直接编译了

[caiduoduo12138]

我pip和conda 都试了，不行
要不你发一下你的安装命令行，我试试，可能我的安装姿势不对
谢谢

[nostayup]

我pip和conda 都试了，不行
要不你发一下你的安装命令行，我试试，可能我的安装姿势不对
谢谢

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

[hhcs9527]

add export TORCH_CUDA_ARCH_LIST="8.0 + PTX"

[lucasjinreal]

@hhcs9527 @nostayup @caiduoduo12138 Pytorch 1.7 was supported on master branch.

Theoretically would also works on lower or higher version. If not pls open a new issue.