duda11.2对应的pytorchvision版本
时间: 2023-08-06 17:07:56 浏览: 79
torch和torchvision版本树莓派python3.7专用
如果您想在 CUDA 11.2 上使用 PyTorch 和 PyTorch Vision,您需要选择适用于 CUDA 11.2 的 PyTorch Vision 版本。根据官方文档,以下是 CUDA 11.2 对应的 PyTorch Vision 版本:
- PyTorch Vision 0.9.1+cu111:适用于 CUDA 11.1 和 CUDA 11.2;
- PyTorch Vision 0.8.2+cu110:适用于 CUDA 11.0 和 CUDA 11.1;
- PyTorch Vision 0.8.1+cu101:适用于 CUDA 10.1 和 CUDA 10.2。
请注意,cu111、cu110 和 cu101 都是 PyTorch Vision 用于标识 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 Vision,您应该选择 PyTorch Vision 0.9.1+cu111 版本。同时,您还需要确保您的操作系统和 GPU 符合 PyTorch Vision 0.9.1+cu111 版本的要求。
阅读全文