pytorch安装教程

目录

1.Anaconda安装

2.CUDA与CuDNN安装

（1）电脑的CUDA版本

（2）官网下载相对应的CUDA:

（3）安装CUDA

（4）安装相对应版本的cuDNN

3.GPU版Pytorch安装

（1）建立python虚拟环境(命名为new,随便命名)

（2）激活环境（在anaconda prompt中输入）

（3）安装pytorch

（4）验证是否安装成功

1.Anaconda安装

（不同的python项目要采取不同的python虚拟环境，不同的虚拟环境之间相互隔离，python版本和包均不共用）

1、官网下载安装包：https://www.anaconda.com/distribution/ 

2、运行并选择安装路径，等待安装完成(要记得勾选 Add Anaconda to the system PATH environment variable，是为了将Anaconda添加到环境变量中）

3、查看是否安装成功

2.CUDA与CuDNN安装

（1）电脑的CUDA版本

桌面右键点击进去NVIDIA控制面板，找到左下角的系统信息

点击组件，出现如下界面。NVCUDA.DLL 这一行后面的CUDA 12.2，说明电脑所支持的最高版本是12.2。

（2）官网下载相对应的CUDA:

网址：CUDA Toolkit Archive | NVIDIA Developer

找到CUDA 12.2版本

选择对应版本下