mlflow 1.30 -> 2.1
This commit is contained in:
8
test_pytorch_m1.py
Normal file
8
test_pytorch_m1.py
Normal file
@ -0,0 +1,8 @@
|
||||
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())
|
Reference in New Issue
Block a user