最最最头疼的就是配环境了
包之间的版本匹配问题
INSTALLING PREVIOUS VERSIONS OF PYTORCH
要考虑到pytorch
和torchvision
之间的匹配关系
显卡版本匹配问题
nvidia-smi
右上角写的CUDA Version: 12.6
表示显卡支持的最高CUDA版本是12.6,后面你安装的版本请不要超过它。
最最最头疼的就是配环境了
INSTALLING PREVIOUS VERSIONS OF PYTORCH
要考虑到pytorch
和torchvision
之间的匹配关系
nvidia-smi
右上角写的CUDA Version: 12.6
表示显卡支持的最高CUDA版本是12.6,后面你安装的版本请不要超过它。