test_mlflow/test_pytorch_m1.py

8 lines
275 B
Python

import torch
import math
print('this ensures that the current MacOS version is at least 12.3+')
print(torch.backends.mps.is_available())
print('\nthis ensures that the current current PyTorch installation was built with MPS activated.')
print(torch.backends.mps.is_built())