1、先查看电脑cuda版本

参考我的另一篇博文第1步:https://blog.csdn.net/s_daqing/article/details/115743576?spm=1001.2014.3001.5501

2、再去pytorch官网查看对应版本关系

https://pytorch.org/get-started/previous-versions/

比如我的选择这里:
在这里插入图片描述
前提先安装conda,参考我的另一篇博文:
https://blog.csdn.net/s_daqing/article/details/117846862

Logo

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

更多推荐