pytorch libtorch 不同版本对应的cuda

官网链接:Previous PyTorch Versions | PyTorch
食用方法:注意低版本的 pytorch 是否支持更高版本的 cuda。(高版本的pytorch一般能兼容低版本cuda)
例如:你需要 1.7.0 的 pytorch,那么 cuda 只能 11.0 及以下(但是,可以变通,你可以在更高版本的cuda下面创建11.0或者更低版本的虚拟环境)。

官方推荐的cuda版本为10.2和11.3,这两种 cuda 支持大多数的 pytorch 版本。

PyTorch 版本 CUDA 环境
0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 9.2
1.2.0、1.1.0、1.0.0(1) 10.0
1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 10.1
1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0 10.2
1.7.0(1) 11.0
1.8.0(1)、1.9.0、1.10.0 11.1
1.8.0(1)、1.9.0、1.10.0 11.3

=========================================================

 

下载地址(下载后离线快速安装):

https://download.pytorch.org/whl/torch_stable.html

安装说明:

https://pytorch.org/get-started/locally/

以前的说明:

Previous PyTorch Versions | PyTorch

参考资料:

 Previous PyTorch Versions | PyTorch

PyTorch Versions · pytorch/pytorch Wiki · GitHub

PyTorch和CUDA版本对应关系 – 墨莲玦 – 博客园

来源:今年不吃饭…

物联沃分享整理
物联沃-IOTWORD物联网 » pytorch libtorch 不同版本对应的cuda

发表评论