pytorch安装及踩坑

pytorch安装很简单，在它官网选择电脑对应的操作系统、package、language和compute platform后，复制其生成的代码，运行后即可安装。因此，安装关键是选择正确的上述参数。操作系统、package和language很简单，按照如下方式确定自己电脑的compute platform：

安装

确定自己电脑有无显卡：

打开“设备管理器”：
查看显示适配器中有无显卡，和显卡版本：

有显卡的电脑，确定cuda版本：

打开NVIDIA控制面板：左下点击“系统信息”，即可查看显卡的驱动程序版本、cuda版本等参数：

安装pytorch

注意：若pytorch官网无cuda正确版本选项（如下），生成代码后注意修改相应版本（11.8改为11.5）：

运行修改后的代码安装即可。

注意坑来了：

解决1：添加conda-forge

无解

解决2：按照报错提示进入anaconda.org中搜索pytorch-cuda=11.5包

但无解，搜不到这个包

解决3：重新检查NVIDIA驱动更新和cuda驱动更新

！！！！！！注意坑又来了，此时（2024.10.23）pytorch支持cuda12.4，但是cuda最新版本是12.6.所以，下载安装cuda时候注意安装12.4.

参考cuda版本卸载和更新进行cuda降版。

验证cuda安装完成：
进入cmd中运行nvcc -V见下得安装完成：

anaconda prompt中运行nvidia-smi见下：

cmd和anaconda prompt中的cuda版本不一样？？？（不影响：）
之后进入pytorch官网选择代码后，到anaconda prompt中运行安装。安装之后验证如下：

True！