拉勾
先说好了,都是我从网上找的,我也没检查过这几张表是不是一样的,也没怎么试过。但是下边的几张表肯定存在错误,因为之前碰到过。记录一下方便查找,所以仅供参考。
1. PyTorch各版本对GPU算力要求
GPU算力:NVIDIA显卡算力
PyTorch版本 | 算力要求 |
>=1.3 | >=3.5 |
>=1.7 | >=5.2 |
2. Torch、Torchvision和Python版本对应关系
3. PyTorch和CUDA版本对应关系
好像是有点问题的,不保证是对的!但总体应该没问题!
4. CUDA和驱动版本对应关系
并不是说安装了 CUDA 之后 torch.cuda.is_available() 返回 True 就能用,能不能跑还得看程序运行报不报错!
5. 总表(PyTorch、Torchvision、Python和CUDA对应关系)
没仔细看,应该有些地方和上边的那几张不是一样的!