Windows下Pytorch入门深度学习环境安装配置GPU版

安装的各块内容联系

步骤 

安装Anaconda 

下载链接： https://www.anaconda.com/

 

可以更换路径：（不要有中文路径）

都打钩

 

利用conda安装pytorch

从开始菜单中打开Anaconda Prompt

输入命令：

conda env list

 查看Anaconda中有哪些虚拟环境：

然后创建一个名字叫Pytorch的虚拟环境：

输入命令：

conda create -n pytorch python=3.9

 

 

自此装好；

激活虚拟环境：

输入命令：

conda activate pytorch 

 查看刚刚安装的虚拟环境里面有啥包：

输入命令：

conda list

 

 

 

如果以后要移除虚拟环境：
输入命令：

conda remove -n pytorch --all 

注意：上面的pytorch是安装的时候的虚拟环境的名称！！！

安装CUDA

 

三个东西一定要版本相互匹配；

第一步确定自己显卡的算力：

先看看自己显卡的版本

 

确定显卡算力：

打开链接：

https://en.wikipedia.org/wiki/CUDA 

 

 

算力是7.5

确定CUDA Runtime版本：

 

10.0-12.3都可以；

然后看看自己的驱动CUDA Driver Version

输入命令：

nvidia-smi 

版本为11.7 

最终确定我们可以安装的CUDA Runtime 版本可以是10.0、10.2、11.0、11.1-11.7.1；

 

推荐使用右面的；

下载安装最新的驱动程序

打开nvidia官网： 

链接：https://www.nvidia.com/en-us/

 

点Drivers：

 

输入自己电脑的类型：

 

 

下载安装即可！ 

更新之后：

 那实际上，更新以后CUDA Runtime版本可以安装最新的12.3的；

 打开pytorch官网：查看：

https://pytorch.org/

 

在pytorch环境中安装：

 

复制官网中的安装命令：

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

 安装即可！

安装完后看看pytorch虚拟环境中的包：

验证Pytorch：

 

 

自此成功！！！