Commit e07192ba authored by Pere Mato Vila's avatar Pere Mato Vila
Browse files

Followup change for SPI-1997

parent 13e85d9a
Pipeline #3144663 passed with stage
in 8 minutes and 10 seconds
......@@ -5402,14 +5402,16 @@ LCGPackage_Add(
LCGPackage_Add(
torch_scatter
URL ${GenURL}/torch_scatter-<VERSION>.tar.gz
ENVIRONMENT LD_LIBRARY_PATH=${torch_home}/lib/python${Python_config_version_twodigit}/site-packages/torch/lib:\$ENV{LD_LIBRARY_PATH}
ENVIRONMENT FORCE_CUDA=1
LD_LIBRARY_PATH=${torch_home}/lib/python${Python_config_version_twodigit}/site-packages/torch/lib:\$ENV{LD_LIBRARY_PATH}
"TORCH_CUDA_ARCH_LIST=6.0 6.1 7.2+PTX 7.5+PTX"
CONFIGURE_COMMAND <VOID>
BUILD_COMMAND <VOID>
INSTALL_COMMAND ${MakeSitePackagesDir}
COMMAND ${PYTHON} setup.py install ${PySetupOptions}
COMMAND ${CMAKE_SOURCE_DIR}/pyexternals/Python_postinstall.sh <INSTALL_DIR>
BUILD_IN_SOURCE 1
DEPENDS torch
DEPENDS torch cuda
)
#---torch_sparse-------------------------------------------------------------------
......@@ -5441,7 +5443,4 @@ LCGPackage_Add(
DEPENDS torch cuda numpy tqdm scipy networkx scikitlearn requests pandas pyparsing PyYAML Jinja2
torch_scatter torch_sparse
)
#'rdflib',
#'googledrivedownloader',
#'yacs',
# Addional dependencies? 'rdflib', 'googledrivedownloader', 'yacs'
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment