pytorch的安装

一、安装

1、第一步，进入终端

终端选择
1、pycharm下的Terminal
2、输入cmd，选择anaconda下的终端

2、第二步，创建虚拟环境

conda create --name 虚拟环境的名字 python=版本

eg：conda create --name pytorch_cpu python=3.7

3、第三步，激活虚拟环境

conda activate 虚拟环境的名字

eg：conda activate pytorch_cpu

4、第四步，输入安装命令

1>torch安装

pip install torch
(pip install torch -i https://mirrors.aliyun.com/pypi/simple/) 我用下来，阿里云这个安装速度更快
如果torch实在安装不成功，试试conda install pytorch

2>torchvision安装

pip install torchvision
(pip install torchvision -i https://mirrors.aliyun.com/pypi/simple/)
如果安装torchvision失败，可能存在问题：要求指定版本的pytorch
pip install torch1.4.0
或conda install pytorch1.4.0
再去装torchvision

5、第五步，验证

显示安装成功后，继续输入：
python
import torch
import torch vision
不报错，即安装成功

二、注意事项

1、pytorch不能安装在32位电脑上
2、建议将pytorch安装在虚拟环境中（其实也可以安装在anaconda里）
3、安装两个模块：torch、torchvision

三、设置编译器

安装好后，在pycharm中的操作：设置编译器
1>新建项目（原项目也行）
2>点File—Settings—项目编译器Python Interpreter
3>点Add
4>点System Interpreter—点右边的…
5>点anaconda3–envs–找刚新建的虚拟环境–点它对应的python.exe

6>然后一路OK就行了

三、安装测试

安装好以后，放入这个py文件，不报错，则环境都整好了

import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torchvision import datasets, transforms
import torchvision
from torch.autograd import Variable
from torch.utils