I need to test a system that has multiple GPUs, linked with NVLink. What is the minimal code I could run to show NVLink actually works?
See this simple code example - how would you change it to take advantage of NVLink?
https://pytorch.org/tutorials/beginner/basics/quickstart_tutorial.html 18