在MacBook Pro M2上使用Tensorflow检测到GPU，但未使用Pytorch检测到GPU

nr7wwzry 于 10个月前发布在 Mac

关注(0)|答案(1)|浏览(158)

我试图找出为什么GPU被识别为：

import tensorflow as tf
tf.test.is_gpu_available() #I'm getting TRUE as output

字符串
而不是：

import torch
torch.cuda.is_available() #I'm getting False as output

型
我关于Pytorch的'pip list'输出命令是：

torch                         2.0.1
torchaudio                    2.0.2
torchvision                   0.15.2

型
如何修复Pytorch，使其能够检测GPU？
谢谢你的好意

1条答案

配备M2的Mac没有支持CUDA的GPU。但是，您可以使用MPS acceleration：

torch.backends.mps.is_available() # True
device = torch.device("mps")
x = torch.tensor([1,2,3], device=device) # This will use MPS acceleration.

字符串