PyTorch与cuda对应版本
时间: 2024-01-10 12:04:08 浏览: 131
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
PyTorch与cuda需要对应版本才能正常运行,可以通过查看PyTorch官方文档或者cuda官方文档来确定对应版本。具体而言,PyTorch版本与cuda版本的对应关系如下:PyTorch 1.3.0对应cuda 10.1,PyTorch 1.2.0对应cuda 10.0,PyTorch 1.1.0对应cuda 9.0,PyTorch 1.0.0对应cuda 9.0,PyTorch 0.4.0对应cuda 8.0。
阅读全文