这个错误提示说明您的PyTorch安装没有启用CUDA,因此无法在GPU上运行模型。要解决此问题,您需要重新安装PyTorch并启用CUDA。

首先,确保您的计算机上已经安装了NVIDIA GPU和CUDA Toolkit。然后,安装符合您CUDA版本的PyTorch。例如,如果您的CUDA版本是10.1,则可以使用以下命令安装:

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/cu101/torch_stable.html

如果您使用的是其他CUDA版本,请将上面的命令中的“cu101”替换为您的CUDA版本。然后再次运行代码,应该就可以在GPU上运行模型了。

Traceback most recent call last File dVscodeDataPythonface_detectfacenet-pytorch-mainpredictpy line 6 in module model = Facenet File dVscodeDataPythonface_detectfacenet-pytorch-mainfacenetpy l

原文地址: https://www.cveoy.top/t/topic/brJw 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录