diff --git a/test/distributed/test_symmetric_memory.py b/test/distributed/test_symmetric_memory.py index c27ad2f10f9ed..ec6aa7f903bf7 100644 --- a/test/distributed/test_symmetric_memory.py +++ b/test/distributed/test_symmetric_memory.py @@ -28,7 +28,9 @@ def requires_cuda_p2p_access(): cuda_p2p_access_available = ( - torch.cuda.is_available() and torch.cuda.device_count() >= 2 + torch.cuda.is_available() + and torch.cuda.get_device_capability() >= (8, 0) + and torch.cuda.device_count() >= 2 ) num_devices = torch.cuda.device_count() for i in range(num_devices - 1):