哪个 PyTorch 版本是正确的？

Question

所以，我已经为 tensorflow keras 下载了 CUDA 11.2，效果很好，但现在我想使用 pytorch 进行强化学习，目前 pytorch 只有两个 CUDA 版本可供 pip 下载，10.2 和 11.3
如果可能的话，我不想升级到 cuda 11.3 或降级到 10.2，如果不这样做，是否可以为不推荐的 cuda 版本下载 pytorch？ 喜欢 10.2 推荐的 pytorch 是否适用于 cuda 11.2 版本

Answer 1

https://github.com/veritas9872/PyTorch-Universal-Docker-Template

请试试这个存储库。 它包含一个模板，用于在任何版本的 CUDA、cuDNN、Python 等上构建任何版本的 PyTorch。

觉得有用就点个star吧！