目的:解决不同项目之间需要的cuda与cudnn 版本可能各不相同的环境问题

第一步:查看自己电脑支持的cuda的版本的最高版本是多少,如我这里12.2 。 那么按照小于12.2即可

第二步:创建虚拟环境并激活

第三步:查找通道里有的cuda与cudnn版本:

如果没用找到需要的版本可以添加通道:

第四步:按照需要的cuda与cudnn版本

如果出现:

说明cudnn已经存在(可能在你前面安装cuda的是时候自带安装了对应的cudnn)

第五步:检查cuda与cudnn是否安装成功

OK 啦

Logo

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

更多推荐