search

haofeixu / aanet

[CVPR'20] AANet: Adaptive Aggregation Network for Efficient Stereo Matching
Apache License 2.0
534 stars 102 forks source link

The problem of cuda? #25

Closed xuzhongyou closed 4 years ago

xuzhongyou commented 4 years ago

The readme writes the cuda version is 10.0 that means the cudatoolkit in pytorch or the cuda version in /usr/local/cuda，i think the build work is by the /usr/local/cuda. i meet the segmentation fault, and not sure the problem comes from cuda version. The cuda version in my path is cuda9.2 and the pytorch environment follow the environment.yaml.

haofeixu commented 4 years ago

Have you successfully installed pytorch with cuda support (you can check the output of running torch.cuda.is_available())? Could you please post more error messages? Thanks.