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的历史版本