windows下安装pytorch GPU版本文章目录***1确定自己的硬件信息确定自己的电脑是否有英伟达## 标题NVIDIA显卡:*****2下载安装Anaconda并进行一些虚拟环境操作******3.接下来是要为安装pytorch做一些关于GPU的准备******3.安装pytorch******4.验证pytorch安装成功*****总结**主要操作1确定自己的硬件信息确定自己的电脑是否有英伟达NVIDIA显卡。2下载Anaconda。3利用conda或者pip安装pytorch(我利用的是pip)安装pytorch的准备1,确定自己安装的cuda版本与GPU的关系 2安装GPU版本pytorch4验证pytorch是否安装成功安装流程*1确定自己的硬件信息确定自己的电脑是否有英伟达## 标题NVIDIA显卡:确定这个很简单你需要打开【任务管理器】中的性能选项卡点击GPU字样的部分就可以查看图片实例如下如图中可以看到NVIDIA字样就说明有英伟达显卡2下载安装Anaconda并进行一些虚拟环境操作官网:https://www.anaconda.com/,进入官网后选择如图所示选择Anaconda Distribution,然后选择跳过登记选择第二张图片的字样直接安装结果如下之后是验证Anaconda是否安装成功首先安装完之后打开Environments中可以看到有一个基础虚拟环境base,如下图接下来是利用Anaconda进行一些虚拟环境的操作下面是一些命令行创建新的虚拟环境conda create -n 名称 python你想创建的版本建议3.8以上方便后面下载pytorch)进入先前创建的虚拟环境conda activate 名称查看虚拟环境中的包conda list查看创建的虚拟环境conda env list删除虚拟环境的所有东西conda remove -n 名称 --all确定cuda driver的版本nvidia-smi实例调用含有Anaconda的命令窗口输入conda env list,可以看到只有一个base在这个基础上创建一个新的虚拟环境名字是xlpytorch,python的版本是3.6建议大家是3.8直接输入命令行conda create -n xlpytorch python3.6(注意空格结果如图所示我还创建了一个新的虚拟环境名字为xxik-xl,python的版本为3.10命令行为conda create -n xxik-xl python3.10,结果如下图所示3.接下来是要为安装pytorch做一些关于GPU的准备确定需要安装的cuda与GPU的关系确定自己电脑的算力这里我们直接安装显卡驱动的最新版本确保自己的CUDA Driver 版本要CUDA Runtime版本以及CUda runtime version cuda driver version,从而确保自己的电脑可以有足够的算力跑代码算力方面可以去cuda官方的网址去找找接下来我们来安装显卡驱动最新版本。网址https://www.nvidia.cn/Download/index.aspx?langcn,进入之后根据自己的GPU来选择最新版本我的电脑如下图所示因为我的是笔记本所以选择的是带有Notebooks的选择之后直接点击查找开始下载我的如下图所示点击下载安装过程中一定要关闭游戏别问我是怎么知道的我的壁纸引擎还没搞好/(ㄒoㄒ)/~~安装过程中的选择默认继续就行安装地址可以自己确定。下图是没有安装前的cuda Driver version(输入命令行nvidia-smi可以查看可以看到有13.0的字样接下来是安装最新版本之后的cuda Driver version可以看到有13.3的字样:3.安装pytorch根据cuda driver的版本我们就可以去pytorch 的官网去确定cuda Runtime版本网址https://pytorch.org/,进入网址后往下滑有安装的字样进而选择适合自己的最新版本因为我的cuda driver为13.3的版本所以我选择的是cuda 13.2的版本pip安装(因为cuda driver version要cuda runtime version)选择完之后我们要在虚拟环境在安装pytorch,先进入先前创建好的虚拟环境然后将安装命令行复制下来按回车键就可以安装了我的命令行是pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu132这个命令行就在选择完版本的下面直接复制就好下面第一个图片是进入先前的虚拟环境第二个图片是安装完成的截图然后我们可以进入这个虚拟环境看一下可以看到torch 和torch vision的字样pypi是指通过pip安装的。4.验证pytorch安装成功相应的一些步骤1.激活对应的虚拟环境你安装的pytorch的虚拟环境命令行conda activate 虚拟环境名2.输入conda list ,看有没有torch3.输入python4.输入import torch5.输入torch.cuda.is_available()6.如果显示True就说明这个pytorch安装成功了总结主要是确定cuda runtime的版本以及进行cuda driver的最新版本安装这是我第一次安装成功将过程分享给大家希望能帮到大家