一般是torch和cuda版本不对
#1 查看cuda版本
nvcc --version
#2 重新安装torch
卸载原先的torch gpu版本
pip uninstall torch torchvision torchaudio
再去官网找正确版本安装

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐