安装历史版本pytorch(含GPU驱动版本查询及与cudatoolkit对应关系)

一、打开cmd,输入nvidia-smi即可查看安装历史版本pytorch(含GPU驱动版本查询及与cudatoolkit对应关系)_第1张图片
驱动升级下载地址:https://www.nvidia.com/Download/index.aspx?lang=en-us

二、查找适合驱动的cudatoolkit版本(见table1):https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

三、安装与cudatoolkit对应的pytorch:https://pytorch.org/get-started/previous-versions/

需要按照以上顺序进行匹配,这样可以避免升级CUDA驱动,从而防止其他虚拟环境因驱动升级而不能使用GPU

你可能感兴趣的:(pytorch,cuda)