diff --git a/chapter1/1.2-pytorch-installation.md b/chapter1/1.2-pytorch-installation.md index b12b7516..c29510f8 100644 --- a/chapter1/1.2-pytorch-installation.md +++ b/chapter1/1.2-pytorch-installation.md @@ -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 @@ -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` 这样才能使用镜像源。