duda11.2对应的pytorch版本
时间: 2023-08-06 17:07:56 浏览: 147
1.10.0GPU版本pytorch环境,conda安装支持cuda10.2和11.3 pip安装支持cuda10.2和11.1
PyTorch 的版本通常与 CUDA 版本和操作系统有关,因此如果您想使用 CUDA 11.2,您需要选择适用于 CUDA 11.2 的 PyTorch 版本。根据官方文档,以下是 CUDA 11.2 对应的 PyTorch 版本:
- PyTorch 1.8.1+cu111:适用于 CUDA 11.1 和 CUDA 11.2;
- PyTorch 1.7.1+cu110:适用于 CUDA 11.0 和 CUDA 11.1;
- PyTorch 1.6.0+cu101:适用于 CUDA 10.1 和 CUDA 10.2。
请注意,cu111、cu110 和 cu101 都是 PyTorch 用于标识 CUDA 版本的缩写。其中,cu111 表示 CUDA 11.1 或 CUDA 11.2,cu110 表示 CUDA 11.0 或 CUDA 11.1,cu101 表示 CUDA 10.1 或 CUDA 10.2。
因此,如果您想在 CUDA 11.2 上使用 PyTorch,您应该选择 PyTorch 1.8.1+cu111 版本。同时,您还需要确保您的操作系统和 GPU 符合 PyTorch 1.8.1+cu111 版本的要求。
阅读全文