安装历史版本pytorch(含GPU驱动版本查询及与cudatoolkit对应关系)
一、打开cmd,输入nvidia-smi即可查看驱动升级下载地址:https://www.nvidia.com/Download/index.aspx?lang=en-us二、查找适合驱动的cudatoolkit版本(见table1):https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html三、安装与cudatoolkit
·
一、打开cmd,输入nvidia-smi即可查看
驱动升级下载地址: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
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
所有评论(0)