import torch print(torch.version.cuda) 10.1 torch.cuda.is_available() /Users/djennelbaroudhadjcheikh/anaconda3/lib/python3.8/site-packages/torch/cuda/init.p