更新:pytorch 1.3的安装说明

This commit is contained in:
zergtant 2019-10-12 20:23:15 +08:00
parent ab8cb1bb69
commit 5f48da7922

View File

@ -1,49 +1,31 @@
# 1.2 Pytorch环境搭建
PyTorch的安装十分简单根据[PyTorch官网](https://pytorch.org/),对系统选择和安装方式等灵活选择即可。
这里以[anaconda](https://www.anaconda.com/)为例,简单的说一下步骤和要点。
国内安装anaconda建议使用[中科大](http://mirrors.ustc.edu.cn/help/anaconda.html)镜像,快的不是一点半点
国内安装anaconda建议使用[清华](https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/)镜像
目前中科大、清华镜像都已经关闭。直接通过prompt下载很慢并且经常会出现HTTPERROR导致下载失败。
所以可以提前下载好压缩包,然后离线下载,这样就不容易出现下载到一半下载失败令人苦恼的情况。
前些日子,由于合规问题中科大、清华镜像都已经关闭。目前只有清华镜像恢复,所以目前可以继续使用
# 1.2.1 安装Pytorch
anaconda安装完成后可以开始创建环境这里以win10 系统为例。打开Anaconda Prompt
pytorch的安装经过了几次变化请大家以官网的安装命令为准。另外需要说明的就是在1.2版本以后pytorch只支持cuda 9.2以上了所以需要对cuda进行升级目前测试大部分显卡都可以用包括笔记本的MX250也是可以顺利升级到cuda 10.1。
我个人测试使用官网的安装命令进行安装时并不能安装1.3版原因未知如果大家conda安装也有问题可以一起讨论下原因所以这里建议大家使用pip进行安装经过测试 pip是没有任何问题的。
```bash
#pytorch为环境名这里创建python3.6版。
conda create -n pytorch python=3.6
# 切换到pytorch环境
activate pytorch
# ***以下为1.0版本安装***
#安装GPU版本根据cuda版本选择cuda80cuda92如果cuda是9.0版,则不需要
#直接conda install pytorch -c pytorch 即可
# win下查看cuda版本命令nvcc -V
conda install pytorch cuda92 -c pytorch
# cpu版本使用
# conda install pytorch-cpu -c pytorch
# torchvision 是torch提供的计算机视觉工具包后面介绍
pip install torchvision
# *** 官方更新了1.01 所以安装方式也有小的变更
# torchversion提供了conda的安装包可以用conda直接安装了
# cuda支持也升级到了10.0
# 安装方式如下:
# cpu版本
conda install pytorch-cpu torchvision-cpu -c pytorch
# GPU版
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
# cudatoolkit后跟着相应的cuda版本
# 目前测试 8.0、9.0、9.1、9.2、10.0都可安装成功
#默认 使用 cuda10.1
pip3 install torch===1.3.0 torchvision===0.4.1 -f https://download.pytorch.org/whl/torch_stable.
#cuda 9.2
pip3 install torch==1.3.0+cu92 torchvision==0.4.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html
#cpu版本
pip3 install torch==1.3.0+cpu torchvision==0.4.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
```
验证输入python 进入
```python
import torch
torch.__version__
# 得到结果'0.4.1'
# 得到结果'1.3.0'
```
## 1.2.2 配置 Jupyter Notebook