cuda和cudnn资源链接:https://pan.baidu.com/s/1hrO6OJuq4dQ1NaYPq7l0pw 
提取码:so8p 

 

经验之谈:

1.安装MXNet gpu 版本(mxnet-cu100),一定要安装好对应版本的cuda和cudnn,一定要对应上,要不然会出现各种问题。

2.打开NVIDA控制面板可查看GPU显卡所支持的CUDA版本,我的是CUDA11.0,表示的是最高支持cuda11.0版本,cuda11.0及以下版本都可以安装。如果你需要安装pytorch的话,建议先查看pytorch版本对应的cuda,然后再下载对应的cuda,因为不同版本的pytorch提供两三个对应cuda版本。

Logo

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

更多推荐