pytorch（GPU版win64）安装全过程（Anaconda+cuda）

反季节探险

已于 2024-01-29 16:59:19 修改

阅读量923

点赞数 7

文章标签： pytorch 人工智能 python

于 2024-01-29 16:54:39 首次发布

本文链接：https://blog.csdn.net/m0_61869170/article/details/135903182

版权

新手第一次安装，踩了很多坑，第一次写记录一下，可能有些乱。

GPU版本要先确定自己有NVIDIA哦。

一、首先是安装Anaconda

在官网下载Anaconda，官网地址：Anaconda | Unleash AI innovation and value

运行安装包

安装路径最好保持为英文

后续就是：next->next->next->finish(大致就是这样，很简单)

环境变量的话是自动添加了，如果需要手动添加的话，可以参考以下的代码，记得把路径"D:\Atool\Anaconda3"改成你们实际路径的地址：

D:\Atool\Anaconda3
D:\Atool\Anaconda3\Scripts
D:\Atool\Anaconda3\Library\bin
D:\Atool\Anaconda3\Library\mingw-w64\bin
D:\Atool\Anaconda3\Library\usr\bin

检查是否安装成功：

同时按 win + r ，输入cmd，打开命令行窗口，查看anaconda版本，输入：

conda --version

查看python版本：

python

出现版本号即代表配置成功。

出现该界面即为安装成功。

二、安装CUDA

这个可以先安装也可以后期再一起安装。

这里需要注意的一点就是版本号，一定要查看一下自己电脑的显卡。

命令行输入：

nvidia-smi

我安装了11.3版本的。

CUDA下载地址（清华镜像源）：CUDA Toolkit 10.1 update1 Archive | NVIDIA Developer

1.先安装

1.我是选择了精简安装，后期环境变量都是配置好的。

2.也可以选择自定义安装。选择“自定义安装”，选择以下四个组件后，点击下一步完成！

如果安装完后没有报错，并且

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

目录路径都存在的话，可进行下一步设置环境变量。

右击“此电脑”，点击属性，点击“高级系统设置”，点击“环境变量”，在“系统变量”中找到“Path”,选中改行，点击“编辑”，再点击“新建”，将下面路径写进去。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\libnvvp

后期再检查一下版本号即可。

nvidia-smi

2.后期安装

在接下来的步骤中会提到。

三、Pytorch下载及安装

以CUDA11.3，python3.8为例：

1）创建虚拟环境

①利用 conda create 指令创建新的虚拟环境

在应用列表里找到Anaconda文件夹，点击anconda prompt，进入命令行，输入以下代码，创建一个虚拟环境。

conda create –n 虚拟环境名字 python=版本

例子中代码应为：(虚拟环境名字可以按自己的意愿取，也可以写为pytorch)

conda create -n py38 python=3.8

②输入y进行下载

③安装完成后，输入

conda info --envs

出现下图所示即为成功。

如果下载慢也可以使用镜像：

conda create –n 虚拟环境名字 python=版本 –c 镜像地址

清华镜像：Index of /anaconda/pkgs/main/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

阿里镜像：

anaconda-pkgs-main安装包下载_开源镜像站-阿里云 (aliyun.com)

2）进入虚拟环境

conda activate pytorch

3）利用conda安装PyTorch

确定了自己的cuda版本号，进入pytorch官网：Start Locally | PyTorch

先找到需要的下载代码：

打开anaconda prompt，并输入

红框处的下载代码。

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

根据提示输入y进行pytorch的安装（可能需要等待，但官网可能一直卡在搜包阶段）。

4）验证

下载完成后，输入

conda list

出现pytorch或torch即为成功。要注意看下安装的版本是不是GPU版本。

5）测试

在该窗口按顺序输入

1	python
2	import torch
3	torch.cuda.is_available()

出现"True"则说明可以使用。

安装完毕喽！！！:)

四、其中遇到的问题（无处不在的bug -_-! ）

1.电脑显卡和CUDA版本匹配的问题

是参考的这篇博客：学习记录：Windows系统cuda11.6，安装pytorch1.12.0、python3.9_cuda11.6对应的pytorch版本-CSDN博客

2.python版本过高（3.11），pytorch不匹配（想降级为3.8）

1）一开始也采用了简单方法：（直接在conda环境降级）

打开cmd，输入

conda install python==3.8

但是没有效果。

2）采用创建一个新的环境

①cmd指令

conda create --name py38 python=3.8

②手动创建

打开：

安装相关包：

conda create -n py38 python=3.8 anaconda

激活环境：

activate py38

然后就实现了python降级（某种意义上的降级 =_= ）

3.使用conda install 时一直卡在solving environment

检查C:\Users\UserName 路径下的.condarc文件

将该行删除即可。（上面的都是可用的镜像网站）

4.用了该方法后下载的直接是2.版本的pytorch与我的python和cuda不匹配，只好卸载pytorch

pip uninstall torch

后来再次安装pytorch，但提示说已经有该版本，便再次查找原因，发现安装后的是CPU版本，遂再次卸载，并查找安装GPU版本的方法。

conda activate py38 # 进入该conda环境
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ # 换conda源
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ # 换conda源
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ # 换conda源
conda install cudatoolkit=11.3 -c http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/ # 安装 cuda
conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/# 安装pytorch