Remember that the K40s do not support the latest PyTorch operations due to their old compute capabilities (e.g., torch.compile).
Remember that the K40s do not support the latest PyTorch operations due to their old compute capabilities (e.g., torch.compile).