CUDA是什么？

CUDA（Compute Unified Device Architecture），是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构（ISA）以及GPU内部的并行计算引擎。 （以上解释来自百度词条）

查看CUDA版本

选择合适的Pytorch

官网链接：https://pytorch.org/

由于我的cuda版本是11.1，且这里显示请用cuda11.3，我们选择下部的Previous，即pytorch的历史版本