记录一下安装pytorch的过程,之前还写过一个《初次安装pytorch过程》,因为换电脑了,所以需要在新电脑上再装一次pytorch,由于是第二次装,所以步骤比第一次更加精简,而且这次是做足了功课之后装的,不像第一次装错版本后又卸载……
推荐一下我看的教程👉最详细的 Windows 下 PyTorch 入门深度学习环境安装与配置 CPU GPU 版 | 土堆教程
如果电脑有Anaconda且知道怎么创建虚拟环境的话,可以直接从这开始看
- 首先需要查看CUDA driver的版本,可以在命令行窗口中输入Nvidia–smi
图中所示部分就是CUDA driver的版本,同时我们要确保自己的CUDA Driver 版本>= CUDA Runtime 版本,我的CUDA driver版本是12.1,所以我可安装<=12.1的CUDA,然后打开pytorch的官网👉pytoch官网