Ubuntu22.04 5080配置深度学习环境

废话不多说，我直接上配置
首先是Ubuntu22.04的系统，5080的显卡，目前(2025年12月11日)只能用最新的cuda128的版本，太低都不能用。
然后用mamba创建python3.11的环境
涉及到的torch和torchvision以及torchaudio版本如下所示

一定要注意，torchvision版本一定要比torch版本晚一天或者两天，在同一天都不行
下面是相关的链接

torchvision-0.25.0.dev20251121+cu128-cp311-cp311-manylinux_2_28_x86_64.whl
torchaudio-2.10.0.dev20251121+cu128-cp311-cp311-manylinux_2_28_x86_64.whl
torch-2.10.0.dev20251120+cu128-cp311-cp311-manylinux_2_28_x86_64.whl

然后运行的效果如下，能够运行d2l里面的resnet。

觉得有用，点个赞再走~

参考链接： https://blog.csdn.net/qq_52810166/article/details/146939649