also set $TORCH_CUDA_ARCH_LIST for PyTorch tests

[Flamefire]

(created using eb --new-pr)

[boegel]

Test report by @boegel

Overview of tested easyconfigs (in order)

• SUCCESS PyTorch-1.3.1-foss-2019b-Python-3.7.4.eb

Build succeeded for 1 out of 1 (1 easyconfigs in total)
node3552.doduo.os - Linux RHEL 8.2, x86_64, AMD EPYC 7552 48-Core Processor (zen2), Python 3.6.8
See https://gist.github.com/7e6fd4141353357046d7517d32e99e69 for a full test report.

[boegel]

Test report by @boegel

Overview of tested easyconfigs (in order)

• SUCCESS PyTorch-1.7.1-foss-2020b.eb

Build succeeded for 1 out of 1 (1 easyconfigs in total)
node3108.skitty.os - Linux centos linux 7.9.2009, x86_64, Intel(R) Xeon(R) Gold 6140 CPU @ 2.30GHz, Python 3.6.8
See https://gist.github.com/22bea9d8cffd265fe0e5fcb7f3911658 for a full test report.

[boegel]

Test report by @boegel

Overview of tested easyconfigs (in order)

• SUCCESS PyTorch-1.3.1-fosscuda-2019b-Python-3.7.4.eb
• SUCCESS PyTorch-1.6.0-fosscuda-2019b-Python-3.7.4-imkl.eb
• SUCCESS PyTorch-1.7.1-fosscuda-2020b.eb

Build succeeded for 3 out of 3 (3 easyconfigs in total)
node3301.joltik.os - Linux centos linux 7.9.2009, x86_64, Intel(R) Xeon(R) Gold 6242 CPU @ 2.80GHz (cascadelake), Python 3.6.8
See https://gist.github.com/37284f9061949eac15e1cf205b2ea5ae for a full test report.

[boegel]

lgtm