一、打开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

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐