Skip to content

Commit

Permalink
更新:更新安装指南增加新版官网安装说明
Browse files Browse the repository at this point in the history
  • Loading branch information
zergtant committed Feb 28, 2019
1 parent fb61056 commit fba6ab2
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions chapter1/1.2-pytorch-installation.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ anaconda安装完成后可以开始创建环境,这里以win10 系统为例。
conda create -n pytorch python=3.6
# 切换到pytorch环境
activate pytorch
# ***以下为1.0版本安装***
#安装GPU版本,根据cuda版本选择cuda80,cuda92,如果cuda是9.0版,则不需要
#直接conda install pytorch -c pytorch 即可
# win下查看cuda版本命令nvcc -V
Expand All @@ -19,6 +20,20 @@ conda install pytorch cuda92 -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都可安装成功


```
需要说明的一点是如果使用清华源,可以直接添加pytorch源镜像去掉,并且去掉`-c pytorch` 这样才能使用镜像源。

Expand Down

0 comments on commit fba6ab2

Please sign in to comment.