[PyTorch] PyTorch安装对各组件版本的要求


拉勾

先说好了,都是我从网上找的,我也没检查过这几张表是不是一样的,也没怎么试过。但是下边的几张表肯定存在错误,因为之前碰到过。记录一下方便查找,所以仅供参考。

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对应关系)

没仔细看,应该有些地方和上边的那几张不是一样的!