深度学习搞如火朝天离不开各大学习框架,更少不了CUDA的计算速度加成buff。各家深度学习框架疯狂推新,cuda也不甘落后,苦了开发者要做各种的版本选择。一不小心就掉坑里了,还半天爬不出来那种。
记录下,以备再掉坑好找根绳子。
Tensorflow
参考 Tensorflow 官网推荐

Pytorch
torch-1.5 | cuda10.0, cuda10.1 |
---|
torch-1.3 | cuda10.0 |
torch-1.2 | cuda10.0 |
torch-1.1 | cuda9.0 |
Cuda && Driver Version

仅供参考。