cuda版本与pytorch版本
时间: 2023-08-08 15:06:00 浏览: 144
Linux Ubuntu 安装 PyTorch 示例
PyTorch-GPU与CUDA版本对应如下:
PyTorch-GPU 1..:CUDA 9.
PyTorch-GPU 1.1.:CUDA 9.、CUDA 10.
PyTorch-GPU 1.2.:CUDA 9.、CUDA 10.、CUDA 10.1
PyTorch-GPU 1.3.:CUDA 9.2、CUDA 10.、CUDA 10.1
PyTorch-GPU 1.4.:CUDA 10.、CUDA 10.1、CUDA 10.2
PyTorch-GPU 1.5.:CUDA 10.1、CUDA 10.2
PyTorch-GPU 1.6.:CUDA 10.1、CUDA 10.2、CUDA 11.
PyTorch-GPU 1.7.:CUDA 10.1、CUDA 10.2、CUDA 11.
PyTorch-GPU 1.8.:CUDA 11.1
PyTorch-GPU 1.9.:CUDA 11.1、CUDA 11.2
需要注意的是,不同版本的PyTorch-GPU可能需要不同版本的CUDA才能正常运行。因此,在安装PyTorch-GPU时,需要根据自己的CUDA版本选择相应的PyTorch-GPU版本。
阅读全文