Skip to content

Commit

Permalink
Update GPU_ARCHS with 75 (#73)
Browse files Browse the repository at this point in the history 
With TRT5 support integrated, I'd propose building with sm_75 by default.
  • Loading branch information
@KellenSunderland
KellenSunderland authored and Yinghai Lu committed Nov 22, 2018
1 parent f1c7aa6 commit a9a47bb
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions CMakeLists.txt
View file
Expand Up @@ -106,6 +106,7 @@ list(APPEND GPU_ARCHS
53
61
70
75
)

set(CUDA_VERBOSE_BUILD ON)
Expand Down

1 comment on commit a9a47bb

@bpinaya
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@KellenSunderland I had issues with this when building on CUDA9 with the driver version 384, had to comment 75 for it to work. On CUDA10 it works with no problems but just to keep if mind if anyone has the same issue.

Please sign in to comment.