这段时间在某站上跟着“跟李沐学AI”博主学习Pytorch,马上学习到神经网络,众所周知,神经网络最好使用GPU来进行运作,但是电脑虽然有GPU但是不能使用。
可以看到,电脑上是有GPU的,但是出现了下述情况
虽然电脑上有GPU,但是却显示GPU数量为0。解决方法如下:
选择对应版本,在命令窗cmd执行命令:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
最终显示成功!