Followup change for SPI-1997

pyexternals/CMakeLists.txt

@@ -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'