Pytorch-gpu环境篇
·
最最最头疼的就是配环境了
包之间的版本匹配问题
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,后面你安装的版本请不要超过它。
更多推荐
所有评论(0)