PYTORCH和CUDA 版本对应关系
时间: 2023-05-25 20:05:56 浏览: 2998
Pytorch-0.4.1兼容CUDA9.0
以下是PyTorch和CUDA版本对应关系的概览:
PyTorch 版本 | CUDA 版本
---|---
1.x.x | 9.0、9.2、10.0、10.1、10.2、11.0、11.1
0.4.x | 8.0、9.0
需要注意的是,不同的PyTorch版本可以支持多个CUDA版本,但是PyTorch和CUDA版本之间存在一定的兼容性问题,建议在使用PyTorch之前检查系统中安装的CUDA版本是否与PyTorch支持的版本匹配,并安装合适的CUDA驱动程序。
阅读全文