install pytorch new

1. Window环境下安装pytorch

网盘搜索代码:86qw84nm

1.1 安装环境

首先你得安装python,这个比较简单,安装一下Anaconda就可以了,
我最近安装的版本是Anaconda 3-5.2.0 / 3-5.3.0,
现在到官网的话,下载的版本是 Python 3.7 version,
Anaconda往期版本下载,清华镜像5.16已经彻底关闭了,
Neo lei | Anaconda 清华源关闭后,怎么办?

install pytorch new_第1张图片
anaconda和python版本对应关系

我已经安装了,可以检查一下自己的python版本。

cmd
C:\Users\Dong>python -V
Python 3.7.0

然后要安装cuda和cudnn,
windows下安装coda相对简单,
现在cuda的版本官方更新到 CUDA Toolkit 10.1,
也可以下载往期的版本,
CUDA Toolkit Archive
我这里的版本:cuda_9.0.176_win10,
安装选择精简,一路下一步默认安装即可,


cudnn
the nvidia cuda deep neural network library,
是面向深度神经网络的GPU加速的原语库,为标准例程提供高度调整的实现,例如前向和后向卷积,池化,规范化和激活层。
这个需要注册账号,9@qq.com,123s*456,
下载时需要和cuda的版本对应,

install pytorch new_第2张图片
cudnn

我安装的版本:cudnn-9.0-windows10-x64-v7.3.1.20
安装方法如下:
tensorflow-gpu安装教程(VS2017+CUDA9.0+cuDNN v7.3.1 for WIN10)

去官网下载CUDNN,下载之前先注册一个账号,选择跟cuda版本对应的cudnn版本。
解压压缩包,分别将下面三个文件拷贝到相应的位置:(是你CUDNN解压后的安装位置)
Copy \cuda\bin\cudnn64_7.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin.
Copy \cuda\ include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include.
Copy \cuda\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64.

安装方法很简单,把压缩包解压,覆盖到cuda安装目录中对应的文件夹中即可。


install pytorch new_第3张图片
cudnn解压并复制到目标文件夹

1.2 安装pytorch

从官网选择自己的安装参数,确定安装指令


install pytorch new_第4张图片
python install

安装 !

这里使用conda的方法,输入一行指令即可进行下载和安装,但是会出现因为网速的问题导致的下载中断而报错,没关系,多进行几次就可以了,你会发现之前没有下载成功的内容,会因为多试几次而下载成功。

注意 !

还要注意一点就是,cmd以管理员的身份运行,

流程如下:

cmd(管理员)
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
# 省略
Downloading and Extracting Packages
pytorch-1.1.0        | 427.2 MB  | ############################################################################ | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
# 如果没有管理员的身份,最后一项,Executing transcaction: failed
# 不过如果下载好了,重新以管理员身份进入cmd,是不需要重新下载的

这部分可忽略不看 !

这里我也尝试了pip的安装方法,报错。两行指令输入分别如下所示。

C:\Users\Dong>pip3 install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl
Collecting torch==1.1.0 from https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl
  Downloading https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl (728.6MB)
     |████████████████████████████████| 728.6MB 13kB/s
Requirement already satisfied: numpy in c:\programdata\anaconda3\lib\site-packages (from torch==1.1.0) (1.15.1)
Installing collected packages: torch
Successfully installed torch-1.1.0
C:\Users\Dong>pip3 install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
Collecting torchvision==0.3.0 from https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
  ERROR: HTTP error 403 while getting https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
  ERROR: Could not install requirement torchvision==0.3.0 from https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl because of error 403 Client Error: Forbidden for url: https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
ERROR: Could not install requirement torchvision==0.3.0 from https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl because of HTTP error 403 Client Error: Forbidden for url: https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl for URL https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl

第二行指令报错的原因应该是因为网址获取出现问题,然后import torch失败,说明第一行指令安装后,torch并不能用。

1.3 测试pytorch安装成功

cmd
C:\Users\Dong>python
Python 3.7.0 (default, Jun 28 2018, 08:04:48) [MSC v.1912 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch

#test-1
>>> x = torch.Tensor([1.0])
>>> xx = x.cuda()
>>> print(xx)
tensor([1.], device='cuda:0')

#test-2 cudnn
>>> from torch.backends import cudnn
>>> print(cudnn.is_acceptable(xx))
True

#test-3
>>> from __future__ import print_function
>>> import torch
>>> x = torch.rand(5,3)
>>> print(x)
tensor([[0.4016, 0.6665, 0.8761],
        [0.4753, 0.5120, 0.4740],
        [0.2295, 0.9175, 0.9715],
        [0.8575, 0.2051, 0.1044],
        [0.5620, 0.6545, 0.5608]])

2. Linux环境下安装pytorch

留坑

你可能感兴趣的:(install pytorch new)