Windows10下Pytorch配置

Posted 2021-02-12 mjhr

主要是通过Anaconda环境下配置Pytorch

一、安装Anaconda

1 进入官网，选择相应版本的Anaconda下载（可能需要不少时间）

2 安装比较简单，基本都是直接next，

3 在Advanced Options环节记得选择加入环境变量（Add Anaconda to my PATH environment variable），不然的话还需要自行配置环境变量

4 安装完成后可以在Windows菜单中找到Anaconda

二、安装pytorch

1 首先进入官网，Get Started，根据自己的配置来查找指令。比如我这里就是选择Windows下使用conda安装，语言按照我电脑配置的python版本选择3.6，CUDA选择None

2 打开Anaconda Prompt（上一步在Windows菜单中看到的那个），不考虑多个python环境的话，直接输入官网给出的指令，运行即可。我这里是：

conda install pytorch-cpu torchvision-cpu -c pytorch

3 可能会遇到HTTP error，并提示retry，这是由于我们采用的镜像源的问题，可以换源采用清华的镜像源：

　将第2步的原指令中的

-c pytorch

　改为

–c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

4 安装完成后可以在Anaconda Prompt中输入

python

import torch

import torchvision

　如果无误则说明安装完成了

三、 Pycharm中配置pytorch

1 虽然我们经过前面的步骤在Anaconda Prompt已经可以无误的import torch，但是在Pycharm中还是会提示没有torch包，这是由于我们的Pycharm还没有接入Anaconda环境

2 在file ->Setting->Project Interpreter ->Add...->System Interpreter->找到Anaconda中的python.exe文件，选择完成后点击Apply即可

3 这里的python.exe的位置因人而异，可以在Anaconda Prompt中输入指令

conda info –envs

查找位置

4 Apply之后Pycharm会进行一段时间的更新，可以在Pycharm最下方看到进度，进度完成后我们的import torch就不会报错了