仅需一步，将cuda、cudnn装到conda虚拟环境中，无需考虑版本，无需考虑cudatoolkit-dev安装不上导致无法利用cuda进行编译
【代码】仅需一步，将cuda、cudnn装到conda虚拟环境中，无需考虑版本，无需考虑cudatoolkit-dev安装不上导致无法利用cuda进行编译。
·
conda activate **你的环境名**
conda install nvidia::cuda
pip3 install torch torchvision torchaudio
完成~
上述conda install是会默认给你安装最新版的cudatoolkit\cudnn还有负责编译的一些文件
如果你想指定版本
可以
conda install nvidia/label/cuda-11.7.1::cuda
欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
3
0- 0
已为社区贡献1条内容
所有评论(0)