{tools}[foss/2025a] torchinfo v1.8.0, torchview v0.2.7 w/ CUDA 12.8.0

easybuild/easyconfigs/t/torchinfo/torchinfo-1.8.0-foss-2025a-CUDA-12.8.0.eb

@@ -0,0 +1,26 @@
+easyblock = 'PythonPackage'
+
+name = 'torchinfo'
+version = '1.8.0'
+versionsuffix = '-CUDA-%(cudaver)s'
+
+homepage = 'https://github.com/tyleryep/torchinfo'
+description = """
+Torchinfo provides information complementary to what is provided by print(your_model) in PyTorch,
+similar to Tensorflow's model.summary() API to view the visualization of the model, which is helpful
+while debugging your network. In this project, we implement a similar functionality in PyTorch and
+create a clean, simple interface to use in your projects.
+"""
+
+toolchain = {'name': 'foss', 'version': '2025a'}
+
+sources = [SOURCE_TAR_GZ]
+checksums = ['72e94b0e9a3e64dc583a8e5b7940b8938a1ac0f033f795457f27e6f4e7afa2e9']
+
+dependencies = [
+    ('Python', '3.13.1'),
+    ('CUDA', '12.8.0', '', SYSTEM),
+    ('PyTorch-bundle', '2.9.1', f'{versionsuffix}-whl'),
+]
+
+moduleclass = 'tools'

easybuild/easyconfigs/t/torchview/torchview-0.2.7-foss-2025a-CUDA-12.8.0.eb

@@ -0,0 +1,27 @@
+easyblock = 'PythonPackage'
+
+name = 'torchview'
+version = '0.2.7'
+versionsuffix = '-CUDA-%(cudaver)s'
+
+homepage = 'https://github.com/mert-kurttutan/torchview'
+description = """
+Torchview provides visualization of pytorch models in the form of visual graphs. Visualization
+includes tensors, modules, torch.functions and info such as input/output shapes.
+"""
+
+toolchain = {'name': 'foss', 'version': '2025a'}
+
+sources = [SOURCE_TAR_GZ]
+checksums = ['c5898fd03d256333c9827000c1f83e1052fbcfc1066032d86b30d429dd942f7c']
+
+builddependencies = [('hatchling', '1.27.0')]
+
+dependencies = [
+    ('Python', '3.13.1'),
+    ('CUDA', '12.8.0', '', SYSTEM),
+    ('PyTorch-bundle', '2.9.1', f'{versionsuffix}-whl'),
+    ('graphviz-python', '0.21'),
+]
+
+moduleclass = 'tools'