实际上是torch版本问题，说明当前的Pytorch版本无法使用显卡
先卸载原先pytorch版本
pip uninstall torch
pip uninstall torchvision然后下载torch 1.9.1版本以及torchvision 0.10.1版本
conda install torch=1.9.1
conda install torchvision=0.10.1解决Torch not compiled with CUDA enabled问题
猜你喜欢
转载自blog.csdn.net/m0_73811793/article/details/129371945
今日推荐
周排行