机器学习环境搭建--cuda相关

CUDA是什么?

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

查看CUDA版本

机器学习环境搭建--cuda相关_第1张图片
机器学习环境搭建--cuda相关_第2张图片

机器学习环境搭建--cuda相关_第3张图片

选择合适的Pytorch

官网链接:https://pytorch.org/
机器学习环境搭建--cuda相关_第4张图片
由于我的cuda版本是11.1,且这里显示请用cuda11.3,我们选择下部的Previous,即pytorch的历史版本
机器学习环境搭建--cuda相关_第5张图片

你可能感兴趣的:(机器学习入门,深度学习,python)