Ubuntu安装pytorch和cuda

为了使用gpu跑深度学习的模型，需要配置环境

nvidia相关驱动

首先是nvidia相关驱动

在官网找到Linux的版本下载

Official Drivers | NVIDIA

下载方法

使用 `wget` 命令来下载一个链接。在终端中输入以下命令：

wget <link>

其中 `<link>` 是你要下载的链接。如果你想将下载的文件保存到指定的目录中，可以使用 `-P` 参数，例如：
wget -P /path/to/directory <link>

这会将下载的文件保存到 `/path/to/directory` 目录中。如果你想指定下载文件的名称，可以使用 `-O` 参数，例如：
wget -O filename <link>

这会将下载的文件保存为 `filename`。如果你想同时使用 `-P` 和 `-O` 参数，可以这样做：
wget -P /path/to/directory -O filename <link>

这会将下载的文件保存到 `/path/to/directory` 目录中，并将其命名为 `filename`。

使用下面命令查看是否安装成功，后者可以实时查看gpu使用情况

nvdia-smi

watch -n 1 nvidia-smi

若出现版本不兼容，可以尝试重启

anaconda

官网下载即可

cuda

CUDA Toolkit 12.1 Update 1 Downloads | NVIDIA Developer

pytorch

Start Locally | PyTorch