Pytorch安装

一、window

下载CUDA Driver

nvidia-driver(NVIDIA驱动程序)是操作系统与NVIDIA GPU硬件之间的软件接口。它负责管理GPU硬件的操作、资源分配、性能优化等任务。CUDA依赖于NVIDIA的驱动程序，因为它需要与GPU硬件进行通信以执行并行计算任务。在使用CUDA进行开发时，确保安装了与CUDA兼容的NVIDIA驱动程序是很重要的。

CUDA12.1是一种并行计算平台和编程模型，而CUDA Toolkit是包含用于CUDA开发的一系列软件工具和库的集合。在进行CUDA开发时，通常需要安装CUDA Toolkit，因为它提供了必要的工具和库，使开发者能够有效地利用NVIDIA GPU的计算资源。

去官网根据显卡类型支持的驱动下载对应的驱动。

 a）win r输入cmd，回车，进入命令行窗口后输入：nvidia-smi查看版本信息

b）如果无效命令需要前往安装驱动