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)