pytorch cuda对应
时间: 2023-09-21 18:09:54 浏览: 170
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
下面是一些常见的 PyTorch 和 CUDA 版本对应关系:
- PyTorch 1.7.0: 支持 CUDA 10.1、CUDA 10.2 和 CUDA 11.0
- PyTorch 1.8.0: 支持 CUDA 10.2、CUDA 11.0 和 CUDA 11.1
- PyTorch 1.9.0: 支持 CUDA 10.2、CUDA 11.0、CUDA 11.1 和 CUDA 11.2
请注意,这些对应关系可能会随着 PyTorch 和 CUDA 的更新而改变。建议在安装 PyTorch 之前,先查看官方文档以获取最新的兼容信息,以确保你选择的 PyTorch 版本与你的 CUDA 版本兼容。
阅读全文