cuda和pytorch版本对应
时间: 2024-06-18 17:05:19 浏览: 202
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
不同版本的PyTorch需要与不同版本的CUDA对应,以保证GPU的正常使用。以下是常见PyTorch版本与CUDA版本的对应关系:
PyTorch v1.0.0对应CUDA 9.0
PyTorch v1.2.0对应CUDA 10.0
PyTorch v1.3.0对应CUDA 10.1
PyTorch v1.4.0对应CUDA 10.1
PyTorch v1.5.0对应CUDA 10.2
PyTorch v1.6.0对应CUDA 10.2
注意,在安装PyTorch时需要选择正确的CUDA版本,否则将会出现运行时错误。
阅读全文