Skip to content
Snippets Groups Projects
Commit c7d801e0 authored by Alexandre Strube's avatar Alexandre Strube
Browse files

Merge branch 'pytorch-geometric-2023-fix-cuda' into '2023'

Add force-cuda and warning about installing from the gpu nodes - already tested and installed

See merge request software-team/easybuild!1925
parents b7c4cd93 bc8b1888
Branches
No related tags found
No related merge requests found
easyblock = 'PythonBundle'
name = 'PyTorch-Geometric'
version = '2.1.0'
local_pytorchver = '1.12.0'
......@@ -14,8 +15,8 @@ dependencies = [
('Python', '3.10.4'),
('CUDA', '11.7', '', SYSTEM),
('PyTorch', local_pytorchver, '-CUDA-%(cudaver)s'),
('scikit-learn', '1.1.2'),
('scikit-image', '0.19.3'),
('scikit-learn', '1.1.2', '', ('gcccoremkl', '11.3.0-2022.1.0')),
('scikit-image', '0.19.3', '', ('gcccoremkl', '11.3.0-2022.1.0')),
('numba', '0.56.4', '-CUDA-%(cudaver)s'),
('h5py', '3.7.0'),
('tqdm', '4.64.0'),
......@@ -26,6 +27,11 @@ dependencies = [
use_pip = True
# This easyblock NEED to be installed from a compute node with gpu,
# see https://github.com/easybuilders/easybuild-easyconfigs/issues/18274
# This here is needed, but not enough:
preinstallopts = "FORCE_CUDA=1"
exts_list = [
('googledrivedownloader', '0.4', {
'modulename': 'google_drive_downloader',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment