Skip to content

{geo}[foss/2023a] TorchGeo v0.6.0 w/ CUDA 12.1.1 #21308

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
branfosj wants to merge 7 commits into from
Closed

{geo}[foss/2023a] TorchGeo v0.6.0 w/ CUDA 12.1.1 #21308

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
4485341
adding easyconfigs: TorchGeo-0.6.0-foss-2023a-CUDA-12.1.1.eb, TorchGe…
branfosj Sep 4, 2024
afdf538
use existing Lightning easyconfigs
branfosj Sep 4, 2024
a97a46a
eb4 not 5
branfosj Sep 4, 2024
bc4bae2
Add CMake builddep for kornia
WilleBell Nov 19, 2024
754aec6
Add CMake builddep for kornia
WilleBell Nov 19, 2024
926d47c
Merge pull request #8 from WilleBell/patch-2
branfosj Mar 4, 2025
5fa3603
Merge branch 'develop' into 20240904084015_new_pr_TorchGeo060
branfosj Mar 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
113 changes: 113 additions & 0 deletions easybuild/easyconfigs/t/TorchGeo/TorchGeo-0.6.0-foss-2023a-CUDA-12.1.1.eb
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,113 @@
# This easyconfig was created by the BEAR Software team at the University of Birmingham.
easyblock = 'PythonBundle'

name = 'TorchGeo'
version = '0.6.0'
versionsuffix = '-CUDA-%(cudaver)s'

homepage = "https://torchgeo.readthedocs.io/"
description = """TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data."""

toolchain = {'name': 'foss', 'version': '2023a'}

builddependencies = [
('Rust', '1.75.0'),
('maturin', '1.4.0', '-Rust-1.75.0'),
('PDM', '2.12.4'),
('CMake', '3.26.3'),
]

dependencies = [
('CUDA', '12.1.1', '', SYSTEM),
('Python', '3.11.3'),
('SciPy-bundle', '2023.07'),
('matplotlib', '3.7.2'),
('PyTorch', '2.1.2', versionsuffix),
('PyTorch-Lightning', '2.2.1', versionsuffix),
('Lightning', '2.2.1', versionsuffix),
('einops', '0.7.0'),
('Fiona', '1.9.5'),
('pyproj', '3.6.0'),
('Rtree', '1.2.0'),
('Shapely', '2.0.1'),
('torchvision', '0.16.0', versionsuffix),
# OPTIONAL
('h5py', '3.9.0'),
('OpenCV', '4.8.1', '%s-contrib' % versionsuffix),
('scikit-image', '0.22.0'),
]

sanity_pip_check = True
use_pip = True

exts_list = [
('kornia_rs', '0.1.5', {
'checksums': ['00520988740e9de1682f2f682caecb9b740d2489f3fbbb1ca0b6fc56af70d298'],
}),
('kornia', '0.7.3', {
'checksums': ['0eb861ea5d7e6c3891ae699a8b7103a5783af0a7c41888ca482420dd3d055306'],
}),
('aenum', '3.1.15', {
'checksums': ['8cbd76cd18c4f870ff39b24284d3ea028fbe8731a58df3aa581e434c575b9559'],
}),
('antlr4-python3-runtime', '4.9.3', { # 4.9.x required for omegaconf and hydra-core
'modulename': 'antlr4',
'checksums': ['f224469b4168294902bb1efa80a8bf7855f24c99aef99cbefc1bcd3cce77881b'],
}),
('omegaconf', '2.3.0', {
'checksums': ['d5d4b6d29955cc50ad50c46dc269bcd92c6e00f5f90d23ab5fee7bfca4ba4cc7'],
}),
('hydra-core', '1.3.2', {
'modulename': 'hydra.core',
'checksums': ['8a878ed67216997c3e9d88a8e72e7b4767e81af37afb4ea3334b269a4390a824'],
}),
('lightly_utils', '0.0.2', {
'checksums': ['a351f3d600f0ab08d12f294725c6457ae000645cb0a1083d0845cb196ccfe698'],
}),
('lightly', '1.5.12', {
'checksums': ['9e98be297057aed51a69837a0cfe526386cf6ac738d22117991cbc95ffc87b4a'],
}),
('huggingface_hub', '0.24.6', {
'checksums': ['cc2579e761d070713eaa9c323e3debe39d5b464ae3a7261c39a9195b27bb8000'],
}),
('safetensors', '0.4.4', {
'checksums': ['5fe3e9b705250d0172ed4e100a811543108653fb2b66b9e702a088ad03772a07'],
}),
('timm', '0.9.7', { # required version for segmentation_models_pytorch
'checksums': ['2bfb1029e90b72e65eb9c75556169815f2e82257eaa1f6ebd623a4b4a52867a2'],
}),
('efficientnet_pytorch', '0.7.1', {
'checksums': ['00b9b261effce59d2d47aae2ad238c29a2a65175470f41ada7ecac439b7c1ee1'],
}),
('pretrainedmodels', '0.7.4', {
'checksums': ['7e77ead4619a3e11ab3c41982c8ad5b86edffe37c87fd2a37ec3c2cc6470b98a'],
}),
('segmentation_models_pytorch', '0.3.4', {
'checksums': ['f4aee7f6add479bd3c3953e855b7055fc657dc6800bf7fc8ab733fd7f8acb163'],
}),
('laspy', '2.5.4', {
'checksums': ['eebdbf3379afbc0b24e7e4812fac567bff880d1e851f70175d22375aaecdf7e1'],
}),
('affine', '2.4.0', {
'checksums': ['a24d818d6a836c131976d22f8c27b8d3ca32d0af64c1d8d29deb7bafa4da1eea'],
}),
('click-plugins', '1.1.1', {
'checksums': ['46ab999744a9d831159c3411bb0c79346d94a444df9a3a3742e9ed63645f264b'],
}),
('cligj', '0.7.2', {
'checksums': ['a4bc13d623356b373c2c27c53dbd9c68cae5d526270bfa71f6c6fa69669c6b27'],
}),
('snuggs', '1.4.7', {
'checksums': ['501cf113fe3892e14e2fee76da5cd0606b7e149c411c271898e6259ebde2617b'],
}),
('rasterio', '1.3.9', { # using easyconfig causes clash in boto* versions with those in PyTorch-Lighning
'use_pip_extras': 'plot,s3',
'checksums': ['fc6d0d290492fa1a5068711cfebb21cc936968891b7ed9da0690c8a7388885c5'],
}),
(name, version, {
'source_tmpl': '%(namelower)s-%(version)s.tar.gz',
'checksums': ['c5b073b3c9ac06cd68e45620bab3a78fb7637fa3563aae4f75f4781ba57aee5a'],
}),
]

moduleclass = 'geo'
112 changes: 112 additions & 0 deletions easybuild/easyconfigs/t/TorchGeo/TorchGeo-0.6.0-foss-2023a.eb
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,112 @@
# This easyconfig was created by the BEAR Software team at the University of Birmingham.
easyblock = 'PythonBundle'

name = 'TorchGeo'
version = '0.6.0'

homepage = "https://torchgeo.readthedocs.io/"
description = """TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data."""

toolchain = {'name': 'foss', 'version': '2023a'}

builddependencies = [
('Rust', '1.75.0'),
('maturin', '1.4.0', '-Rust-1.75.0'),
('PDM', '2.12.4'),
('CMake', '3.26.3'),
]

dependencies = [
('Python', '3.11.3'),
('SciPy-bundle', '2023.07'),
('GDAL', '3.7.1'),
('matplotlib', '3.7.2'),
('PyTorch', '2.1.2'),
('PyTorch-Lightning', '2.2.1'),
('Lightning', '2.2.1'),
('einops', '0.7.0'),
('Fiona', '1.9.5'),
('pyproj', '3.6.0'),
('Rtree', '1.2.0'),
('Shapely', '2.0.1'),
('torchvision', '0.16.0'),
# OPTIONAL
('h5py', '3.9.0'),
('OpenCV', '4.8.1', '-contrib'),
('scikit-image', '0.22.0'),
]

sanity_pip_check = True
use_pip = True

exts_list = [
('kornia_rs', '0.1.5', {
'checksums': ['00520988740e9de1682f2f682caecb9b740d2489f3fbbb1ca0b6fc56af70d298'],
}),
('kornia', '0.7.3', {
'checksums': ['0eb861ea5d7e6c3891ae699a8b7103a5783af0a7c41888ca482420dd3d055306'],
}),
('aenum', '3.1.15', {
'checksums': ['8cbd76cd18c4f870ff39b24284d3ea028fbe8731a58df3aa581e434c575b9559'],
}),
('antlr4-python3-runtime', '4.9.3', { # 4.9.x required for omegaconf and hydra-core
'modulename': 'antlr4',
'checksums': ['f224469b4168294902bb1efa80a8bf7855f24c99aef99cbefc1bcd3cce77881b'],
}),
('omegaconf', '2.3.0', {
'checksums': ['d5d4b6d29955cc50ad50c46dc269bcd92c6e00f5f90d23ab5fee7bfca4ba4cc7'],
}),
('hydra-core', '1.3.2', {
'modulename': 'hydra.core',
'checksums': ['8a878ed67216997c3e9d88a8e72e7b4767e81af37afb4ea3334b269a4390a824'],
}),
('lightly_utils', '0.0.2', {
'checksums': ['a351f3d600f0ab08d12f294725c6457ae000645cb0a1083d0845cb196ccfe698'],
}),
('lightly', '1.5.12', {
'checksums': ['9e98be297057aed51a69837a0cfe526386cf6ac738d22117991cbc95ffc87b4a'],
}),
('huggingface_hub', '0.24.6', {
'checksums': ['cc2579e761d070713eaa9c323e3debe39d5b464ae3a7261c39a9195b27bb8000'],
}),
('safetensors', '0.4.4', {
'checksums': ['5fe3e9b705250d0172ed4e100a811543108653fb2b66b9e702a088ad03772a07'],
}),
('timm', '0.9.7', { # required version for segmentation_models_pytorch
'checksums': ['2bfb1029e90b72e65eb9c75556169815f2e82257eaa1f6ebd623a4b4a52867a2'],
}),
('efficientnet_pytorch', '0.7.1', {
'checksums': ['00b9b261effce59d2d47aae2ad238c29a2a65175470f41ada7ecac439b7c1ee1'],
}),
('pretrainedmodels', '0.7.4', {
'checksums': ['7e77ead4619a3e11ab3c41982c8ad5b86edffe37c87fd2a37ec3c2cc6470b98a'],
}),
('segmentation_models_pytorch', '0.3.4', {
'checksums': ['f4aee7f6add479bd3c3953e855b7055fc657dc6800bf7fc8ab733fd7f8acb163'],
}),
('laspy', '2.5.4', {
'checksums': ['eebdbf3379afbc0b24e7e4812fac567bff880d1e851f70175d22375aaecdf7e1'],
}),
('affine', '2.4.0', {
'checksums': ['a24d818d6a836c131976d22f8c27b8d3ca32d0af64c1d8d29deb7bafa4da1eea'],
}),
('click-plugins', '1.1.1', {
'checksums': ['46ab999744a9d831159c3411bb0c79346d94a444df9a3a3742e9ed63645f264b'],
}),
('cligj', '0.7.2', {
'checksums': ['a4bc13d623356b373c2c27c53dbd9c68cae5d526270bfa71f6c6fa69669c6b27'],
}),
('snuggs', '1.4.7', {
'checksums': ['501cf113fe3892e14e2fee76da5cd0606b7e149c411c271898e6259ebde2617b'],
}),
('rasterio', '1.3.9', { # using easyconfig causes clash in boto* versions with those in PyTorch-Lighning
'use_pip_extras': 'plot,s3',
'checksums': ['fc6d0d290492fa1a5068711cfebb21cc936968891b7ed9da0690c8a7388885c5'],
}),
(name, version, {
'source_tmpl': '%(namelower)s-%(version)s.tar.gz',
'checksums': ['c5b073b3c9ac06cd68e45620bab3a78fb7637fa3563aae4f75f4781ba57aee5a'],
}),
]

moduleclass = 'geo'
Loading