torch找不到CUDA的问题

合集 - torch(9)

1.torch神经网络--线性回归2024-10-052.pytorch之hub模块2024-10-053.torch--minst手写体识别2024-10-064.torch神经网络温度预测2024-10-06

5.torch找不到CUDA的问题2024-12-17

6.torch--fashion mnist drop out2024-11-077.torch--模型选择-欠拟合-过拟合2024-11-068.torch--多层感知机实现图像分类2024-11-049.torch--fashion-minst分类2024-10-31

收起

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)