Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

修复NET_PHASE=2时画面显示为九宫格问题 #15

Closed
LeoJhonSong opened this issue Jul 28, 2020 · 1 comment
Closed

修复NET_PHASE=2时画面显示为九宫格问题 #15

LeoJhonSong opened this issue Jul 28, 2020 · 1 comment

Comments

@LeoJhonSong
Copy link
Owner

现在显示为:
image.png
应该为:
image.png

@LeoJhonSong
Copy link
Owner Author

libtorch版本 (1.5) 比开发时的1.1高导致的. 具体什么原因不清楚.

而libtorch1.1只支持到cuda10.0, 不支持cuda10.2, 因此降级cuda. 而虽然在这之前装的cudnn7.6.5支持cuda10.0, 但既然cudnn官方下载页面上专门有for cuda10.0cuda10.2两个cudnn7.6.5, 应该是对于不同cuda版本, 同一版本的cudnn也有区别, 所以重装cudnn7.6.5.

然后因为libtorch1.1+opencv4会导致一些cv方法未定义的报错, 将opencv版本也降至3.4.

最后, 编译包含cuda支持的opencv需要使用cuda支持的gcc版本 (此处有个各系统gcc版本支持列表). 我一开始用系统默认的gcc-10编译报错了, 类似这样(里面的版本号不一样):

error -- unsupported GNU version! gcc 4.5 and up are not supported!

总结下来:
libtorch: 1.1
cuda: 10.0
cudnn: 7.6.5 (其他支持cuda10.0的版本应该也可以)
opencv: 3.4
编译opencv用gcc: 7

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants