1、查看CUDA版本
-->打开cmd
-->查看CUDA版本 nvidia-smi![](https://img2024.cnblogs.com/blog/3437857/202412/3437857-20241217115657975-986317706.jpg)
2、找到CUDA版本对应的torch版本,并安装
https://pytorch.org/get-started/previous-versions/![](https://img2024.cnblogs.com/blog/3437857/202412/3437857-20241217115856459-346556157.jpg)