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 清华源关闭后,怎么办?
我已经安装了,可以检查一下自己的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的版本对应,
我安装的版本: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安装目录中对应的文件夹中即可。
1.2 安装pytorch
从官网选择自己的安装参数,确定安装指令
安装 !
这里使用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
留坑