标签：8.1 11.1 11.2 cuDNN PyTorch cuda nvidia CUDA

PyTorch当前支持的CUDA只支持到11.1，你可以通过命令 sudo nvidia-smi查看你当前驱动支持的cuda版本，这里我没法确定版本需不需要强制一定匹配，也没有测试。

 我这里按照强制一定匹配的思路来，CUDA 11.1对应的驱动程序是nvidia-driver-455，但是我在ubuntu20上安装这个驱动，总是报失败。

 后来看到国外一个大神，安装nvidia-driver-460驱动，其对应的cuda版本是11.2，这个大神说cuda-11.2也可以在pytorch上使用，请参考他的博文

https://medium.com/analytics-vidhya/install-cuda-11-2-cudnn-8-1-0-and-python-3-9-on-rtx3090-for-deep-learning-fcf96c95f7a1

标签：8.1,11.1,11.2,cuDNN,PyTorch,cuda,nvidia,CUDA
来源： https://blog.csdn.net/humanbeng/article/details/120154434

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。