From fba6ab225bd1976d492526376eefb14a5d33efee Mon Sep 17 00:00:00 2001 From: zergtant Date: Thu, 28 Feb 2019 20:51:17 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=EF=BC=9A=E6=9B=B4=E6=96=B0?= =?UTF-8?q?=E5=AE=89=E8=A3=85=E6=8C=87=E5=8D=97=E5=A2=9E=E5=8A=A0=E6=96=B0?= =?UTF-8?q?=E7=89=88=E5=AE=98=E7=BD=91=E5=AE=89=E8=A3=85=E8=AF=B4=E6=98=8E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- chapter1/1.2-pytorch-installation.md | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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` 这样才能使用镜像源。