torch版本要跟哪几个库对应来
时间: 2023-09-03 20:05:51 浏览: 310
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
要使用哪些库取决于你要使用的模型和功能。通常,PyTorch的版本与CUDA和cuDNN的版本有关。例如,如果你想使用PyTorch 1.8.1,那么你需要符合以下要求的CUDA和cuDNN版本:
- CUDA 11.1
- cuDNN 8.0.5
你可以在PyTorch官方文档中找到更多关于版本兼容性的详细信息。另外,你还需要根据你的操作系统选择正确的PyTorch安装方式。例如,对于Windows,你可以使用pip进行安装,而对于Linux,你可以使用conda或pip进行安装。
阅读全文