当前位置:   article > 正文

Ubuntu安装pytorch和cuda_linux pytorch cuda12.2

linux pytorch cuda12.2

为了使用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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/536787
推荐阅读
相关标签
  

闽ICP备14008679号