Win10安装 pytorch【CPU版】
- 一、安装Anaconda
- 二、安装Pytorch-CPU
-
- 1. 打开 Anaconda Prompt
- 2. 查看conda环境
- 3. 进入 pytorch 环境,安装pytorch
- 4. 到[pytorch官网](https://pytorch.org/get-started/locally/)
- 5. 手动安装方法
- 6. 验证是否安装成功
- 7. vscode配置【笔者是用vscode的】
- 8. 下载 torchvision
一、安装Anaconda
Anaconda安装见 win10安装Tensorflow-cpu(AMD显卡) && keras
二、安装Pytorch-CPU
1. 打开 Anaconda Prompt
输入 下面的命令,创建一个新的屋子【隔离不同版本的环境】
conda create -n pytorch python=3.6
2. 查看conda环境
输入 命令 查看 新建的 pytorch 环境
conda info --envs
其中,右边的 * 号表示,当前处于哪个环境。
3. 进入 pytorch 环境,安装pytorch
输入 命令 ,进入 pytorch 环境,安装pytorch
activate pytorch
命令行开头显示 pytorch【即蓝框】,代表成功进入 pytorch 环境
4. 到pytorch官网
PyTorch Build 选择 Stable;选择系统;Package,Windows下推荐 conda,Linux 下推荐 pip;然后选择 python,如果没有显卡的话,选择 CPU。
我的选择如图所示:
复制下面的代码,接着前面的命令行。
注:如果重新打开 Anaconda Prompt时,查看最左边括号中是 base 还是 pytorch,确保 一定要在 pytorch 环境中。
之后粘贴命令执行即可。
此处最好可以 fq ,或者将镜像源修改为国内镜像源地址,否则会下载很慢甚至出错。而且最好不要切换到其他界面,看到不会动可以按按回车(我就是这样做的hhh,笔者fq了,所以会稍微快了点~)
当pytorch无法获得资源时,需要手动的去下载相关压缩文件,并用conda install 安装。详细步骤如下:
5. 手动安装方法
该方法来自 PyTorch安装最全流程
具体方法如下:进入下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
首先根据个人计算机版本,选择不同的入口。
点击进入,然后通过快捷键找到pytorch,并下载。【好像没有最新版的】,这里以 下载 pytorch-1.4.0-cpu_py36ha775e86_0.tar.bz2 为例。
将下载的压缩包放入Anaconda安装文件夹下对应的pkgs文件夹下。
随后在Prompt中输入:
conda install D:\anaconda\pkgs\pytorch-1.4.0-cpu_py36ha775e86_0.tar.bz2
这里需要注意的是一定要在(pytorch)这个虚拟房间下,运行安装。
其他无法获得镜像安装的可以参照上述离线方法安装。
6. 验证是否安装成功
- 确保已在 pytorch 环境下,输入
python
- 输入
import torch
,如果没有报错,说明 PyTorch 已成功安装上了 - 查看 PyTorch 版本,输入
print(torch.__version__)
- 当然你可以查看你的PyTorch是否可以使用 GPU,输入
torch.cuda.is_available()
若是 True,则可以使用 GPU;是 False,则只能使用CPU。
7. vscode配置【笔者是用vscode的】
添加新的python解释器,
vscode 中 Python 解析器信息显示于左下角,如果没配置会有警告标示,如上图所示:可单击该区域【左下角】,选择一个合适的 Python 解释器,解释器可以是虚拟环境。如上图所示:当正确选择后,就会显示 。
pytorch环境的路径可在 conda环境下查看 ,如
找到对应的路径后,选择 python.exe
在vscode测试下:
8. 下载 torchvision
torchvision 是PyTorch中专门用来处理图像的库。这个包中有四个大类。
- torchvision.datasets:用来进行数据加载的
- torchvision.models:提供了已经训练好的模型
- torchvision.transforms:提供了一般的图像操作类
- torchvision.utils
相应的版本和支持的 Python 版本
可使用下面的命令下载 torchvision
Anaconda:
conda install torchvision -c pytorch
Pip:
pip install torchvision
在Anaconda操作如下图:
最后可通过命令 conda list
查看已下载的包
至此,pytorch-cpu版成功安装,开始pytorch之旅~