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 python=3.6(注意空格),结果如图所示:
我还创建了一个新的虚拟环境名字为xxik-xl,python的版本为3.10,命令行为conda create -n xxik-xl python=3.10,结果如下图所示:
3.接下来是要为安装pytorch做一些关于GPU的准备:
确定需要安装的cuda与GPU的关系:确定自己电脑的算力(这里我们直接安装显卡驱动的最新版本),确保自己的CUDA Driver 版本要>=CUDA Runtime版本,以及CUda runtime version <= cuda driver version,从而确保自己的电脑可以有足够的算力跑代码,算力方面可以去cuda官方的网址去找找,接下来我们来安装显卡驱动最新版本。
网址:https://www.nvidia.cn/Download/index.aspx?lang=cn,进入之后根据自己的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 ,看有没有torch
3.输入python
4.输入import torch
5.输入torch.cuda.is_available()
6.如果显示True,就说明这个pytorch安装成功了
总结:
主要是确定cuda runtime的版本,以及进行cuda driver的最新版本安装,这是我第一次安装成功,将过程分享给大家,希望能帮到大家