Pytorch在安装的过程中,不同版本依赖不同的底层环境,这些包括cuda、cudnn等,用户在安装时,主要考虑cuda版本和pytorch版本的对应关系即可。
1. cuda12.1 - pytorch 2.2.x, 2.1.x
2. cuda11.8 - pytorch 2.2.x, 2.1.x, 2.0.x
3. cuda11.3 - pytorch 1.10.x, 1.9.x, 1.8.1
4. cuda11.1 - pytorch 1.10.0, 1.9.x, 1.8.x
5. cuda11.0 - pytorch 1.7.x
6. cuda10.2 - pytorch 1.10.x, 1.9.x, 1.8.x, 1.7.x,
7. cuda10.1 - pytorch 1.7.x, 1.6.0, 1.5.x, 1.4.0
8. cuda10.0 - pytorch 1.2.0, 1.1.0, 1.0.x
9. cuda9.2 - pytorch 1.7.x, 1.6.0, 1.5.x, 1.4.0, 1.2.0
10. cuda9.0 - pytorch 1.1.0, 1.0.x
11. cuda8.0 - pytorch 1.0.0