我所使用的Ubuntu22.04安装的驱动为nvidia-550
所支持的最高cuda版本为12.4,
但是截至2024年6月30日，pytorch官网最新支持的cuda版本为12.1.
这里如果不手动安装CUDA12.1，选择在安装pytorch时自动安装CUDA，会造成安装驱动525，造成linux驱动崩溃，导致重启电脑无法进入桌面。
注意：已不能选择deb安装！！！，不然会自动安装驱动525，同样出现上面问题。
（哎。重装两遍系统的深刻教训）
正确做法：选择用runfile安装，安装的时候不要选驱动（安装时会有选项，要使驱动前边的方框为空白）。
