Pytorch cuda版本选择(高效简洁版)
Pytorch cuda版本选择 只需要低于cuda驱动版本即可，cuda版本查看是nvidia-smi
·
简而言之
Pytorch cuda版本选择 只需要低于cuda驱动版本即可，cuda驱动版本查看命令是nvidia-smi, nvcc -V 是runtimeapi版本可以不用管
1.只要看cuda驱动版本
安装pytorch 选择cuda版本，只要看你电脑cuda驱动版本即可。
2.选择依据
pytorch中cuda版本只要不高于你电脑cuda驱动版本即可
3.cuda驱动版本方法
nvidia-smi
简单解释
1.原因是cuda 两个版本
一个是驱动版本(nvidia-smi可查)，另一个是runtime api版本（nvcc -v 查的），
2.pytorch的cuda版本只需要低于你的cuda驱动版本即可
欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)