zl程序教程

您现在的位置是:首页 >  硬件

当前栏目

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

机器学习 环境 -- 搭建 相关 CUDA
2023-09-27 14:25:42 时间

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的历史版本
在这里插入图片描述