Pytorch、Tensorflow与CUDA的版本对应

cuda	Pytorch	Tensorflow
10.0	1.0.0~1.4.0	1.13.0~2.0.0
10.1	1.3.0~1.8.1	2.1.0~2.3.0
10.2	1.5.0~1.8.1	无
11.0	1.7.0~1.7.1	2.4.0
11.1	1.8.0~1.9.0	无
11.2	无	2.5.0

• 更新于2021-09-18
• 仅记录cuda10.0以上版本
• Pytorch似乎可以支持低版本的cuda，Tensorflow对版本要求更严格，必须要有对应版本的cudart64_xx.dll（xx为版本号）
• 吐槽一下辣鸡的Tensorflow，竟然直接就不支持cuda10.2了，在官方github的issue中有不少人提到这个问题，官方的回答就是让你去下旧版本或者新版本（呕
• 接上，也有大佬找到了Linux系统下将cuda10.2 symlink 到10.1的方法，见issue链接、解决教程

参考链接

• Pytorch：https://download.pytorch.org/whl/torch_stable.html
• Tensorflow：https://tensorflow.google.cn/install/source_windows