Pytorch1.0安装

2019-02-14  本文已影响0人  Latet

前言

本文记录了在windows下配置pytorch环境的步骤。如需转载,请附出处https://www.jianshu.com/p/07abfe5f82e1

1、Anaconda安装

Anaconda安装,去官网下载比较慢,所以选取清华源镜像链接下的安装包,链接:Anaconda安装包下载
安装好之后打开anaconda navigator,效果如下:

anaconda.png
其中最重要的就是查看环境(environments),最先安装好时只会有一个base环境,我安装的是基于python3.6的base环境的anaconda
conda包管理命令
## 创建指定python3.6版本的环境,环境名为py36
conda create –n py36 python=3.6
## 查看安装的环境名
conda info --envs
## 激活环境名为py36的环境
activate py36

查看安装环境效果:
带星号的代表当前所处环境
图中显示一共有三个环境,root代表默认环境,另外两个代表新建环境

2、pytorch安装(CPU)

pytorch主页提供了安装的命令操作,根据是否有GPU和cuda版本,安装具体版本的pytorch都做了显示。


但是,在安装pytorch和一些基本库(numpy、matplotlib等)时,有些时候网速会特别慢,这个时候安装会采用设置清华源镜像路径来安装。
### 设置清华源镜像
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
 conda config --set show_channel_urls yes
### 设置pytorch镜像
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

添加镜像后将官方安装的命令-c pytorch去掉会快的飞起

 pip install -i https://pypi.tuna.tsinghua.edu.cn/simple name

另外,老版本的镜像中不存在最新版的pytorch1.0,所以采取镜像安装pytorch显示来源也是pytorch官网,针对未翻墙的用户还是特别忙。当然,最适用的办法是下载离线包,然后再进行安装。离线包下载链接

1.png
当然,也可以根据上面的下载链接里的badges里的链接来设置pytorch镜像链接。下载好的离线安装包直接采用conda install name就好了
如下图,安装成功!

3、pytorch安装(GPU)

conda install pytorch cuda91
pip install torchvision

安装检查:显示cuda可用,pytorch版本1.0


上一篇下一篇

猜你喜欢

热点阅读