cuda pytorch最稳定的版本匹配
时间: 2023-07-25 17:24:56 浏览: 143
torch.cuda.is_available()返回False的问题解决
对于 CUDA 和 PyTorch 的版本匹配,以下是一些稳定的组合:
- CUDA 9.2 + PyTorch 0.4.1
- CUDA 10.0 + PyTorch 1.0
- CUDA 10.1 + PyTorch 1.1
- CUDA 10.2 + PyTorch 1.2
- CUDA 11.0 + PyTorch 1.7
- CUDA 11.1 + PyTorch 1.8
注意,这只是一些稳定的组合,其他的版本组合也可能可以正常工作,但是可能会遇到一些问题。因此,为了确保稳定性和兼容性,建议使用上述组合中的一个。
阅读全文