intuit / Trapheus

This tool automates restoration of RDS database instances from snapshots into any dev, staging or production environments. It supports individual RDS Snapshot as well as cluster snapshot restore operations.
https://intuit.github.io/Trapheus/
MIT License
101 stars 52 forks source link

Adding support for CUDA #197

Closed stationeros closed 7 months ago

stationeros commented 7 months ago

Description

Adding support for CUDA

To build the image: docker build -t Dockefile.cuda .

Run the conatiner docker run --gpus all -p 8501:8501 Dockefile.cuda

--gpus all will enable gpu support for the container

Thanks for contributing this Pull Request. Make sure that you submit this Pull Request against the master branch of this repository, add a brief description, and tag the relevant issue(s) and PR(s) below.