torch找不到CUDA的问题

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)
posted @ 2024-12-17 11:59  星空28  阅读(98)  评论(0)    收藏  举报