来来回回在三个系统上都安装了一遍pytorch,其实很简单,先安装显卡驱动,再安装CUDA,最后安装pytorch就好。

pytorch 的安装需要去官网根据配置需要复制对应的下载指令。(推荐 conda 新建一个pytorch环境安装)

教程已经写得很好了,就记录一些自己踩的坑吧

显卡驱动非 OPEN

安装系统recommend 的 open 版本驱动比如我的是 525-open 后,如果检测不到的话,可以换成对应版本的非 open 其驱动。
它是 Linux 的当然推荐开源了

环境变量配置在自己的终端文件中

太蠢了,当时,只知道一步一步跟着教程走。
比如我使用的是 zsh
配置文件就应该放在 vim ~/. zshrc 而不是死死的按照教程的 vim ~/.bashrc
bash - shopt command not found in .bashrc after shell updation - Stack Overflow

验证是否安装成功

验证是否启用了 cuda 加速。如果启用了的话运行结果显然应该是 true。

import torch  
torch.cuda.is_available()

参考链接

Ubuntu搭建Pytorch,就这一篇就够了\_ubuntu pytorch\_Chengyunlai的博客-CSDN博客

新装UBUNTU 22.04安装Nvidia GPU版Pytorch完整教程 | GTrush的个人博客