Cuda、Tensorflow、Pytorch版本匹配

Tensorflow和Cuda、Cudnn版本对应关系

image-20200728202135462

所以Cuda 10.0以上版本,就需要tensorflow 1.13.1以上

Pytorch和Cuda版本对应关系

https://pytorch.org/get-started/previous-versions/

一般就Cuda 10.0的话,pytorch得是1.0.0以上


Cuda、Tensorflow、Pytorch版本匹配
http://yoursite.com/2020/02/10/ML框架/Cuda、Tensorflow、Pytorch版本匹配/
作者
Wei Lyu
发布于
2020年2月10日
许可协议