Updating CUB version in Dockerfile

amazon-archives / amazon-dsstne

Deep Scalable Sparse Tensor Network Engine (DSSTNE) is an Amazon developed library for building Deep Learning (DL) machine learning (ML) models

Apache License 2.0

4.41k stars 731 forks source link

Closed spacelover1 closed 1 year ago

spacelover1 commented 4 years ago

Issue #, if available: #227

Description of changes: Update Dockerfile to download new version CUB library because old CUB library uses legacy CUDA shuffle API.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.