Windows10下基于Anaconda安装PyTorch

所需条件：

1.Anaconda环境

2.安装CUDA（有Nvidia显卡就安装，没有则不用）

安装流程

笔者电脑已经有Anaconda，有Nvidia MX250显卡，但没有CUDA，故先安装CUDA

1、下载CUDA

链接如下：CUDA Toolkit 11.5 Update 1 Downloads | NVIDIA Developer

根据自己的环境选择对应版本，笔者win10选择如下，接下来就是下载文件无脑安装了，可以更改安装路径，但仍有一部分文件是会被安装到C盘，条件允许情况下建议默认路径安装即可，笔者C盘内存告急故选择D盘，接下来，根据流程一步步走，选择精简版安装即可。

2、安装PyTorch

注意，安装之前记得先把Anaconda的源换掉，否则下载很慢或者根本下载不了。过程如下：

2.1创建.condarc文件

Start | Anaconda3 | Anaconda Prompt运行 conda config命令，系统将会在用户的家目录创建.condarc文件，该文件为隐藏文件，窗口查看或者打开的时候，需要设置显示隐藏文件夹才能看到，如图：

2.2修改.condarc文件

打开文件，笔者使用的是Sublime Text

将文件中所有东西替换成如下部分：

channels:
  - https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
 
show_channel_urls: true

然后保存关闭

然后关掉Anaconda Prompt，再重新启动，使更新后的源生效。

2.3安装PyTorch

搜索PyTorch进入官网找到install进入选择对应版本进行安装

也可直接点击链接打开：PyTorch安装，笔者windows10系统为例，点击install之后，默认选择情况如图所示：

打开Anaconda Prompt，运行conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch等待安装即可，若遇到下图所示情况，不要慌，回车即可。

3、验证

3.1验证PyTorch安装是否成功

Anaconda Prompt下运行python

运行以下代码：

·```

import torch
x = torch.rand(5, 3)
print(x)

如图所示

即为安装成功

3.2验证PyTorch安装是否成功

检查 GPU 驱动程序和 CUDA 是否已启用且可由 PyTorch 访问，运行以下命令以返回是否启用了 CUDA 驱动程序：

import torch
torch.cuda.is_available()

perfect！！！

参考文献：

CUDA下载安装：CUDA Toolkit 11.5 Update 1 Downloads | NVIDIA Developer

Anaconda换源：windows10系统，anaconda更换源_moshiyaofei的博客-CSDN博客_anaconda win10换源

PyTorch安装：PyTorch安装

写在最后

如有问题请联系笔者

博客园：https://www.cnblogs.com/ChangciWei/

邮箱:w954514378816172@163.com

​ ——云初