PyTorch 版本与 CUDA 版本的兼容性示例

• PyTorch 1.9.0 及以上版本支持 CUDA 11.1。
• PyTorch 1.8.0 支持 CUDA 11.0。
• PyTorch 1.7.0 支持 CUDA 10.2。
• PyTorch 1.6.0 支持 CUDA 10.1。
• PyTorch 1.5.0 支持 CUDA 10.1。
• PyTorch 1.4.0 支持 CUDA 10.1。
• PyTorch 1.3.0 支持 CUDA 10.0。
• PyTorch 1.2.0 支持 CUDA 9.2。
• PyTorch 1.1.0 支持 CUDA 9.0。

请确保你的 PyTorch 版本与你的 CUDA 版本兼容。如果你需要使用特定的 CUDA 版本，建议安装与之兼容的 PyTorch 版本，以确保顺利使用 GPU 加速功能。

你可以在 PyTorch 官方文档中查找特定版本的 CUDA 兼容性信息，以确保你选择的 PyTorch 版本支持你的 CUDA 版本。