废话不多说，我直接上配置
首先是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