Konstantin Baierer kba
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| moved to https://github.com/mets/METS1to2 |
bertsky
/ cudatest.sh
Last active
June 10, 2023 23:51
OCR-D workflow for coverage tests, esp. CUDA support
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| set -e | |
| # select first CUDA device (in case there are multiple, which may fail due to [a recent Tensorflow problem](https://github.com/qurator-spk/eynollah/issues/99)) | |
| export CUDA_VISIBLE_DEVICES=0 | |
| # check we are not running into [this bug](https://github.com/shapely/shapely/issues/1598) | |
| python3 -c "from shapely.geometry import Polygon; import torch; torch.randn(10).cuda()" | |
| # validate CUDA support is working in TF and Torch (not an exhaustive test) | |
| python3 -c "import torch; print(torch.cuda.is_available())" |
OlderNewer