最最最头疼的就是配环境了

包之间的版本匹配问题

INSTALLING PREVIOUS VERSIONS OF PYTORCH
要考虑到pytorchtorchvision之间的匹配关系
在这里插入图片描述

显卡版本匹配问题

nvidia-smi
在这里插入图片描述
右上角写的CUDA Version: 12.6表示显卡支持的最高CUDA版本是12.6,后面你安装的版本请不要超过它。

更多推荐