python - 哪个 PyTorch 版本是正确的？

Question

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

标签： pythonpytorch

Answer 1

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

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

觉得有用就star一下吧！